ONE - On-device Neural Engine
|
#include <CircleShapeInference.h>
Definition at line 38 of file CircleShapeInference.h.
|
final |
Definition at line 40 of file CircleAdd.cpp.
References luci::sinf::broadcast_shape(), luci::sinf::circle_shape(), luci::CircleAdd::x(), and luci::CircleAdd::y().
|
final |
Definition at line 84 of file CircleBatchMatMul.cpp.
References luci::CircleBatchMatMul::adj_x(), luci::CircleBatchMatMul::adj_y(), luci::sinf::broadcast_shape(), luci::sinf::circle_shape(), INTERNAL_EXN, loco::Dimension::known(), output_shape, luci::CircleBatchMatMul::x(), and luci::CircleBatchMatMul::y().
|
final |
Definition at line 43 of file CircleConcatenation.cpp.
References loco::NodeShape::as(), luci::CircleConcatenation::axis(), INTERNAL_EXN_V, luci::CircleNode::name(), luci::CircleConcatenation::numValues(), output_shape, loco::TensorShape::rank(), luci::shape_get(), and luci::CircleConcatenation::values().
|
final |
Definition at line 40 of file CircleDiv.cpp.
References luci::sinf::broadcast_shape(), luci::sinf::circle_shape(), output_shape, luci::CircleDiv::x(), and luci::CircleDiv::y().
|
final |
Definition at line 46 of file CircleFullyConnected.cpp.
References luci::sinf::circle_shape(), loco::TensorShape::dim(), luci::CircleFullyConnected::input(), luci::CircleFullyConnected::keep_num_dims(), LUCI_ASSERT, loco::TensorShape::rank(), loco::Dimension::unset(), loco::Dimension::value(), and luci::CircleFullyConnected::weights().
|
final |
Definition at line 79 of file CircleIfOut.cpp.
|
final |
Definition at line 33 of file CircleLogistic.cpp.
References luci::sinf::circle_shape(), and luci::CircleLogistic::x().
|
final |
Definition at line 39 of file CircleMul.cpp.
References luci::sinf::broadcast_shape(), luci::sinf::circle_shape(), luci::CircleMul::x(), and luci::CircleMul::y().
|
final |
Definition at line 32 of file CircleNeg.cpp.
References luci::sinf::circle_shape(), and luci::CircleNeg::x().
|
inlinefinalvirtual |
Default fallback.
Reimplemented from luci::CircleNodeVisitor< loco::TensorShape >.
Definition at line 42 of file CircleShapeInference.h.
References loco::NodeShape::as(), and luci::CircleShapeInferenceRule::infer().
|
final |
Definition at line 33 of file CirclePad.cpp.
References luci::sinf::circle_shape(), luci::CirclePad::input(), luci::sinf::pad_shape(), and luci::CirclePad::paddings().
|
final |
Definition at line 33 of file CircleQuantize.cpp.
References luci::sinf::circle_shape(), and luci::CircleQuantize::input().
|
final |
Definition at line 35 of file CircleRange.cpp.
References luci::CircleRange::delta(), GET_RANGE_PARAM, INTERNAL_EXN, luci::CircleRange::limit(), output_shape, and luci::CircleRange::start().
|
final |
Definition at line 70 of file CircleReshape.cpp.
References luci::sinf::circle_shape(), loco::TensorShape::dim(), luci::CircleReshape::Shape::dim(), INFO, INTERNAL_EXN, loco::Dimension::known(), LOGGER, LUCI_ASSERT, luci::CircleReshape::newShape(), output_shape, loco::TensorShape::rank(), luci::CircleReshape::Shape::rank(), luci::sinf::S32, loco::Dimension::set(), luci::CircleReshape::shape(), luci::CircleReshape::tensor(), loco::Dimension::unset(), luci::VALID, and loco::Dimension::value().
|
final |
Definition at line 33 of file CircleRsqrt.cpp.
References luci::sinf::circle_shape(), and luci::CircleRsqrt::x().
|
final |
Definition at line 36 of file CircleSoftmax.cpp.
References luci::sinf::circle_shape(), and luci::CircleSoftmax::logits().
|
final |
Definition at line 385 of file CircleStridedSlice.cpp.
References luci::CircleStridedSlice::begin(), begin, luci::sinf::BuildStridedSliceParams(), luci::sinf::circle_shape(), luci::sinf::StridedSliceContext::effective_input_shape, luci::CircleStridedSlice::end(), luci::CircleStridedSlice::input(), luci::input_node(), INTERNAL_EXN, LUCI_ASSERT, output_shape, loco::TensorShape::rank(), luci::sinf::S32, luci::sinf::StartForAxis(), luci::sinf::StopForAxis(), and luci::CircleStridedSlice::strides().