apollo::perception::inference::Layer< Dtype > Class Template Reference

#include <layer.h>

Inheritance diagram for apollo::perception::inference::Layer< Dtype >:

Collaboration diagram for apollo::perception::inference::Layer< Dtype >:

Public Member Functions
virtual void	ForwardGPU (const std::vector< std::shared_ptr< base::Blob< Dtype >>> &bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &top)=0
virtual void	ForwardCPU (const std::vector< std::shared_ptr< base::Blob< Dtype >>> &bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &top)=0
	Layer ()=default
virtual	~Layer ()=default
void	set_gpu_id (const int &gpu_id)

Protected Attributes
int	gpu_id_ = 0

Constructor & Destructor Documentation

Layer()

template<typename Dtype >

apollo::perception::inference::Layer< Dtype >::Layer ( )

default

~Layer()

template<typename Dtype >

virtual apollo::perception::inference::Layer< Dtype >::~Layer ( )

virtualdefault

Member Function Documentation

ForwardCPU()

template<typename Dtype >

virtual void apollo::perception::inference::Layer< Dtype >::ForwardCPU ( const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  top  )

pure virtual

Implemented in apollo::perception::inference::ROIPoolingLayer< Dtype >.

ForwardGPU()

template<typename Dtype >

virtual void apollo::perception::inference::Layer< Dtype >::ForwardGPU ( const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  top  )

pure virtual

Implemented in apollo::perception::inference::ROIPoolingLayer< Dtype >.

set_gpu_id()

template<typename Dtype >

void apollo::perception::inference::Layer< Dtype >::set_gpu_id ( const int &  gpu_id )

inline

Member Data Documentation

gpu_id_

template<typename Dtype >

int apollo::perception::inference::Layer< Dtype >::gpu_id_ = 0

protected

---

The documentation for this class was generated from the following file:

• modules/perception/inference/layer.h