Public Member Functions | |
InputLayer (op::Operation< T > *input) | |
virtual std::vector< op::Operation< T > * > | get_weights () |
![]() | |
virtual op::Operation< T > * | out () |
op::Operation< T > * | get_input () |
op::Operation< T > * | get_output () |
std::vector< unsigned int > | get_input_shape () const |
std::vector< unsigned int > | get_output_shape () const |
unsigned int | get_input_shape (unsigned int i) const |
unsigned int | get_output_shape (unsigned int i) const |
void | set_name (std::string name) |
std::string | get_name () const |
Protected Member Functions | |
void | init () |
![]() | |
Layer (std::vector< unsigned int > input_shape, op::Operation< T > *input) | |
Additional Inherited Members | |
![]() | |
std::vector< unsigned int > | input_shape |
std::vector< unsigned int > | output_shape |
op::Operation< T > * | input |
op::Operation< T > * | output |
std::string | name |