ONE - On-device Neural Engine
|
#include <Tensor.h>
Public Member Functions | |
Shape (int rank) | |
Shape (std::initializer_list< int32_t > dims) | |
int | num_dims () const |
int32_t | dim (int i) const |
int32_t & | dim (int i) |
int32_t | num_elements () const |
int64_t | large_num_elements () const |
bool | operator== (const Shape &other) const |
bool | operator!= (const Shape &other) const |
|
inlineexplicit |
|
inline |
|
inline |
|
inline |
Definition at line 41 of file Tensor.h.
Referenced by luci_interpreter::kernels::calcOffset(), luci_interpreter::kernels::calculateShapeForBroadcast(), luci_interpreter::kernels::ArgMax::configure(), luci_interpreter::kernels::AveragePool2D::configure(), luci_interpreter::kernels::BatchToSpaceND::configure(), luci_interpreter::kernels::Concatenation::configure(), luci_interpreter::kernels::Conv2D::configure(), luci_interpreter::kernels::DepthToSpace::configure(), luci_interpreter::kernels::DepthwiseConv2D::configure(), luci_interpreter::kernels::FullyConnected::configure(), luci_interpreter::kernels::Gather::configure(), luci_interpreter::kernels::L2Pool2D::configure(), luci_interpreter::kernels::MaxPool2D::configure(), luci_interpreter::kernels::MirrorPad::configure(), luci_interpreter::kernels::Pack::configure(), luci_interpreter::kernels::Pad::configure(), luci_interpreter::kernels::PadV2::configure(), luci_interpreter::kernels::ResizeBilinear::configure(), luci_interpreter::kernels::ResizeNearestNeighbor::configure(), luci_interpreter::kernels::Select::configure(), luci_interpreter::kernels::SpaceToBatchND::configure(), luci_interpreter::kernels::SpaceToDepth::configure(), luci_interpreter::kernels::Split::configure(), luci_interpreter::kernels::SplitV::configure(), luci_interpreter::kernels::Squeeze::configure(), luci_interpreter::kernels::SVDF::configure(), luci_interpreter::kernels::Tile::configure(), luci_interpreter::kernels::Transpose::configure(), luci_interpreter::kernels::TransposeConv::configure(), luci_interpreter::kernels::UnidirectionalSequenceLSTM::configure(), luci_interpreter::kernels::Unpack::configure(), luci_interpreter::kernels::testing::dequantizeTensorData(), luci_interpreter_pal::eval_integer_8x8_16_lstm(), luci_interpreter::kernels::lstm::EvalFloat(), luci_interpreter::kernels::MirrorPad::execute(), luci_interpreter::kernels::Transpose::execute(), luci_interpreter::kernels::getTensorShape(), large_num_elements(), luci_interpreter::kernels::testing::makeInputTensor(), and num_elements().
|
inline |
|
inline |
Definition at line 39 of file Tensor.h.
Referenced by luci_interpreter::kernels::calculateShapeForBroadcast(), luci_interpreter::kernels::ArgMax::configure(), luci_interpreter::kernels::BatchToSpaceND::configure(), luci_interpreter::kernels::BroadcastTo::configure(), luci_interpreter::kernels::Concatenation::configure(), luci_interpreter::kernels::Conv2D::configure(), luci_interpreter::kernels::DepthwiseConv2D::configure(), luci_interpreter::kernels::ExpandDims::configure(), luci_interpreter::kernels::FullyConnected::configure(), luci_interpreter::kernels::Gather::configure(), luci_interpreter::kernels::Mean::configure(), luci_interpreter::kernels::MirrorPad::configure(), luci_interpreter::kernels::OneHot::configure(), luci_interpreter::kernels::Pack::configure(), luci_interpreter::kernels::Pad::configure(), luci_interpreter::kernels::PadV2::configure(), luci_interpreter::kernels::PRelu::configure(), luci_interpreter::kernels::ReduceMax::configure(), luci_interpreter::kernels::ReduceProd::configure(), luci_interpreter::kernels::RmsNorm::configure(), luci_interpreter::kernels::Select::configure(), luci_interpreter::kernels::SpaceToBatchND::configure(), luci_interpreter::kernels::Split::configure(), luci_interpreter::kernels::SplitV::configure(), luci_interpreter::kernels::Squeeze::configure(), luci_interpreter::kernels::StridedSlice::configure(), luci_interpreter::kernels::Sum::configure(), luci_interpreter::kernels::SVDF::configure(), luci_interpreter::kernels::Transpose::configure(), luci_interpreter::kernels::UnidirectionalSequenceLSTM::configure(), luci_interpreter::kernels::Unpack::configure(), luci_interpreter::kernels::testing::dequantizeTensorData(), luci_interpreter::kernels::lstm::EvalFloat(), luci_interpreter::kernels::MirrorPad::execute(), luci_interpreter::kernels::Pad::execute(), luci_interpreter::kernels::PadV2::execute(), luci_interpreter::kernels::StridedSlice::execute(), luci_interpreter::kernels::getTensorShape(), and luci_interpreter::kernels::testing::makeInputTensor().
|
inline |
Definition at line 53 of file Tensor.h.
References dim().
Referenced by luci_interpreter::kernels::FullyConnected::configure(), luci_interpreter::kernels::Mean::configure(), luci_interpreter::kernels::ReduceMax::configure(), luci_interpreter::kernels::ReduceProd::configure(), luci_interpreter::kernels::Sum::configure(), luci_interpreter::kernels::UnidirectionalSequenceLSTM::configure(), luci_interpreter::kernels::testing::dequantizeTensorData(), luci_interpreter::kernels::ExpandDims::execute(), luci_interpreter::kernels::If::execute(), luci_interpreter::kernels::MirrorPad::execute(), luci_interpreter::kernels::Reshape::execute(), luci_interpreter::kernels::testing::makeInputTensor(), luci_interpreter::Tensor::readData(), and luci_interpreter::Tensor::writeData().
|
inline |
|
inline |
Definition at line 74 of file Tensor.h.
Referenced by operator!=().