clDNN
|
#include <concatenation.hpp>
Public Types | |
enum | concatenation_axis { along_b = cldnn_concatenation_along_b, along_f = cldnn_concatenation_along_f, along_x = cldnn_concatenation_along_x, along_y = cldnn_concatenation_along_y } |
typedef cldnn_concatenation_desc | dto |
Public Member Functions | |
concatenation (const primitive_id &id, const std::vector< primitive_id > &input, const concatenation_axis axis, const padding &output_padding=padding()) | |
concatenation (const dto *dto) | |
Constructs a copy from C API (depth_concatenate) | |
![]() | |
const cldnn_primitive_desc * | get_dto () const override |
Returns pointer to a C API primitive descriptor casted to cldnn_primitive_desc. | |
![]() | |
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_id > | dependecies () 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 | |
concatenation_axis | axis |
Dimension along which concatenation should take place. | |
![]() | |
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. | |
Additional Inherited Members | |
![]() | |
primitive_base (const primitive_id &id, const std::vector< primitive_id > &input, const padding &output_padding=padding()) | |
primitive_base (const cldnn_concatenation_desc *dto) | |
![]() | |
virtual std::vector< std::reference_wrapper< const primitive_id > > | get_dependencies () const |
![]() | |
primitive_id_arr | _input |
Concatenation is used to concatenate multiple sources into one destination along specified dimension.
Definition at line 53 of file concatenation.hpp.
|
inline |
id | This primitive id. |
input | Vector of input primitives ids. |
axis | Selected dimension for concatenation. |
Definition at line 69 of file concatenation.hpp.