SpaceToDepth permutes input tensor blocks of spatial data into depth dimension. More...

Inheritance diagram for ngraph::op::v0::SpaceToDepth:

Collaboration diagram for ngraph::op::v0::SpaceToDepth:

Public Types
enum	SpaceToDepthMode { BLOCKS_FIRST, DEPTH_FIRST }

Public Member Functions
const NodeTypeInfo &	get_type_info () const override

	SpaceToDepth (const Output< Node > &data, const SpaceToDepthMode &mode, std::size_t block_size=1)
	Constructs a SpaceToDepth operation. More...

	SpaceToDepth (const Output< Node > &data, const std::string &mode, std::size_t block_size=1)

bool	visit_attributes (AttributeVisitor &visitor) override

std::size_t	get_block_size () const

SpaceToDepthMode	get_mode () const

virtual OutputVector	decompose_op () const override

virtual std::shared_ptr< Node >	clone_with_new_inputs (const OutputVector &new_args) const override

Static Public Attributes
static constexpr NodeTypeInfo	type_info {"SpaceToDepth", 0}

Detailed Description

SpaceToDepth permutes input tensor blocks of spatial data into depth dimension.

Note

Values from the height and width dimensions are moved to the depth dimension.

   Output node produces a tensor with shape:
   [N, C * blocksize * blocksize, H / blocksize, W / blocksize]

Constructor & Destructor Documentation

ngraph::op::v0::SpaceToDepth::SpaceToDepth	(	const Output< Node > &	data,
		const SpaceToDepthMode &	mode,
		std::size_t	block_size = `1`
	)

Constructs a SpaceToDepth operation.

Parameters

data	- Node producing the input tensor
mode	Specifies how the output depth dimension is gathered from block coordinates and the old depth dimension.
block_size	- the size of the block of values to be moved

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