ONE - On-device Neural Engine
Loading...
Searching...
No Matches
loco::TensorShape Class Reference

#include <TensorShape.h>

Public Member Functions

 TensorShape ()=default
 
 TensorShape (std::initializer_list< Dimension > dims)
 
uint32_t rank (void) const
 
void rank (uint32_t r)
 
const Dimensiondim (uint32_t axis) const
 
Dimensiondim (uint32_t axis)
 

Detailed Description

Definition at line 28 of file TensorShape.h.

Constructor & Destructor Documentation

◆ TensorShape() [1/2]

loco::TensorShape::TensorShape ( )
default

◆ TensorShape() [2/2]

loco::TensorShape::TensorShape ( std::initializer_list< Dimension dims)
inline

Definition at line 32 of file TensorShape.h.

32: _dims(dims.begin(), dims.end()) {}

Member Function Documentation

◆ dim() [1/2]

Dimension & loco::TensorShape::dim ( uint32_t  axis)
inline

Definition at line 39 of file TensorShape.h.

39{ return _dims.at(axis); }

◆ dim() [2/2]

const Dimension & loco::TensorShape::dim ( uint32_t  axis) const
inline

Definition at line 38 of file TensorShape.h.

38{ return _dims.at(axis); }

Referenced by loco::NodeShape::as(), moco::as_feature_shape(), moco::as_tensor_shape(), onert::backend::acl_common::asTensorShape(), luci::sinf::BuildStridedSliceParams(), luci::cal_offset(), luci::sinf::circle_shape(), loco::element_count(), luci::get_channel_dim_index(), locoex::COpShapeInferenceRule::infer(), luci::is_same_shape(), moco::new_const(), loco::operator<<(), loco::operator==(), loco::operator==(), luci::sinf::pad_shape(), luci::compute::DepthwiseConv2D::prepare(), luci::compute::FullyConnected::prepare(), luci::CircleShapeInferencePass::run(), luci::DynamicBatchToSingleBatchPass::run(), loco::NodeShape::set(), loco::PermutingDecoder< Domain::DepthwiseFilter >::shape(), loco::PermutingDecoder< Domain::Feature >::shape(), loco::PermutingDecoder< Domain::Filter >::shape(), loco::PermutingDecoder< Domain::Matrix >::shape(), loco::PermutingEncoder< Domain::Feature >::shape(), loco::PermutingEncoder< Domain::Filter >::shape(), loco::PermutingEncoder< Domain::DepthwiseFilter >::shape(), loco::PermutingEncoder< Domain::Matrix >::shape(), moco::TensorPackEnumerator::start(), luci::sinf::StartForAxis(), luci::sinf::StopForAxis(), moco::tensor_shape(), luci::compute::tflite_shape(), exo::to_shape_description(), luci::to_shape_description(), moco::valid_shape_for_constfold_binary_op(), onert::compiler::ShapeValidator::visit(), luci::sinf::Algorithm::visit(), luci::sinf::Algorithm::visit(), and moco::window_of().

◆ rank() [1/2]

void loco::TensorShape::rank ( uint32_t  r)
inline

Definition at line 36 of file TensorShape.h.

36{ _dims.resize(r); }

◆ rank() [2/2]


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