clDNN
cldnn::roi_pooling Struct Reference
Inheritance diagram for cldnn::roi_pooling:
Collaboration diagram for cldnn::roi_pooling:

Public Types

typedef cldnn_roi_pooling_desc dto
 

Public Member Functions

 roi_pooling (const primitive_id &id, const primitive_id &input_data, const primitive_id &input_rois, pooling_mode mode, int pooled_width, int pooled_height, float spatial_scale, int group_sz=0, const padding &output_padding=padding())
 
 roi_pooling (const dto *dto)
 
- Public Member Functions inherited from cldnn::primitive_base< roi_pooling, cldnn_roi_pooling_desc >
const cldnn_primitive_descget_dto () const override
 Returns pointer to a C API primitive descriptor casted to cldnn_primitive_desc.
 
- Public Member Functions inherited from cldnn::primitive
 primitive (const primitive_type_id &type, const primitive_id &id, const std::vector< primitive_id > &input, const padding &output_padding=padding())
 
 primitive (const cldnn_primitive_desc *dto)
 Constructs a copy from basic C API cldnn_primitive_desc.
 
std::vector< std::reference_wrapper< primitive_id > > dependecies ()
 Returns references to all primitive ids on which this primitive depends - inputs, weights, biases, etc.
 
std::vector< primitive_iddependecies () const
 Returns copy of all primitive ids on which this primitive depends - inputs, weights, biases, etc.
 
 operator primitive_id () const
 Implicit conversion to primiitive id.
 

Static Public Member Functions

static primitive_type_id type_id ()
 

Public Attributes

pooling_mode mode
 
int pooled_width
 
int pooled_height
 
float spatial_scale
 
int group_sz
 
- Public Attributes inherited from cldnn::primitive
const primitive_type_id type
 Primitive's type id.
 
const primitive_id id
 Primitive's id.
 
fixed_size_vector_ref input
 List of ids of input primitives.
 
padding output_padding
 Requested output padding.
 

Protected Member Functions

void update_dto (dto &dto) const override
 
- Protected Member Functions inherited from cldnn::primitive_base< roi_pooling, cldnn_roi_pooling_desc >
 primitive_base (const primitive_id &id, const std::vector< primitive_id > &input, const padding &output_padding=padding())
 
 primitive_base (const cldnn_roi_pooling_desc *dto)
 
- Protected Member Functions inherited from cldnn::primitive
virtual std::vector< std::reference_wrapper< const primitive_id > > get_dependencies () const
 

Additional Inherited Members

- Protected Attributes inherited from cldnn::primitive
primitive_id_arr _input
 

Detailed Description

Definition at line 33 of file roi_pooling.hpp.


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