ShapeInference.h File Reference

#include "Utils.h"
#include "ir/operation/Concat.h"
#include "ir/operation/Conv2D.h"
#include "ir/operation/DepthwiseConv2D.h"
#include "ir/operation/Pool2D.h"
#include "ir/operation/Reshape.h"
#include "ir/operation/StridedSlice.h"
#include "compiler/LoweredGraph.h"
#include "ir/Index.h"
#include "ir/OperationVisitor.h"
#include "backend/ITensor.h"
#include "backend/ITensorRegistry.h"

Go to the source code of this file.

Data Structures
struct	onert::shape_inference::StridedSliceParams

Namespaces
namespace	onert
namespace	onert::shape_inference

Typedefs
using	onert::shape_inference::Shapes = std::vector< ir::Shape >

Functions
ir::Shape	onert::shape_inference::inferArgMinMaxShape (const ir::Shape &input_shape, int axis, int rank)
ir::Shape	onert::shape_inference::inferBatchMatMulShape (const ir::Shape &lhs_shape, const ir::Shape &rhs_shape, const ir::operation::BatchMatMul::Param &param)
ir::Shape	onert::shape_inference::inferBCQFullyConnectedShape (const ir::Shape &in_shape, const ir::Shape &cluster_shape, const int32_t *cluster_buf)
ir::Shape	onert::shape_inference::inferBCQGatherShape (const ir::Shape &indices_shape, const ir::Shape &cluster_shape, const int32_t *cluster_buf, int rank, const ir::operation::BCQGather::Param &param)
ir::Shape	onert::shape_inference::inferBroadcastToShape (const ir::Shape shp_shape, const int32_t *shp_buf)
ir::Shape	onert::shape_inference::inferConcatShape (const Shapes &in_shapes, const ir::operation::Concat::Param &param)
ir::Shape	onert::shape_inference::inferConv2DShape (const ir::Shape &in_shape, const ir::Shape &ker_shape, const ir::operation::Conv2D::Param &param)
ir::Shape	onert::shape_inference::inferDepthwiseConv2DShape (const ir::Shape &in_shape, const ir::Shape &ker_shape, const ir::operation::DepthwiseConv2D::Param &param)
ir::Shape	onert::shape_inference::inferEltwiseShape (const ir::Shape &lhs_shape, const ir::Shape &rhs_shape)
ir::Shape	onert::shape_inference::inferExpandDimsShape (const ir::Shape &in_shape, int32_t axis)
template<typename T >
ir::Shape	onert::shape_inference::inferFillShape (const ir::Shape &fill_shape, const T *shape_buf)
ir::Shape	onert::shape_inference::inferFullyConnectedShape (const ir::Shape &in_shape, const ir::Shape &ker_shape)
ir::Shape	onert::shape_inference::inferGatherShape (const ir::Shape &input_shape, const ir::Shape &indices_shape, int axis, int rank)
ir::Shape	onert::shape_inference::inferOnehotShape (const ir::Shape &input_shape, const int depth, int axis)
ir::Shape	onert::shape_inference::inferPackShape (const ir::Shape &input_shape, int axis, int rank, int num)
ir::Shape	onert::shape_inference::inferPadShape (const ir::Shape &in_shape, const int32_t *pad_buf, const size_t num_pads)
ir::Shape	onert::shape_inference::inferPoolShape (const ir::Shape &in_shape, const ir::operation::Pool2D::Param &param)
template<typename T >
ir::Shape	onert::shape_inference::inferRangeShape (T start_val, T limit_val, T delta_val)
ir::Shape	onert::shape_inference::inferReshapeShape (const ir::Shape &input_shape, const int32_t *shape_buf, const int32_t shape_num_elements)
ir::Shape	onert::shape_inference::inferReduceShape (const ir::Shape &input_shape, const std::vector< int > &axes, bool keep_dims)
template<float * >
ir::Shape	onert::shape_inference::inferRangeShape (float *start_val, float *limit_val, float *delta_val)
ir::Shape	onert::shape_inference::inferResizeBilinearShape (const ir::Shape &in_shape, const int32_t output_height, const int32_t output_width)
ir::Shape	onert::shape_inference::inferSelectShape (const ir::Shape &input_cond_shape, const ir::Shape &input_true_shape, const ir::Shape &input_false_shape)
template<typename T >
ir::Shape	onert::shape_inference::inferSliceShape (const ir::Shape &input_shape, const T *begins_buf, const T *sizes_buf)
ir::Shape	onert::shape_inference::inferSpaceToBatchNDShape (const ir::Shape &input_shape, const ir::Shape &block_shape_shape, const ir::Shape &padding_shape, const int32_t *block_shape_buf, const int32_t *padding_buf)
ir::Shape	onert::shape_inference::inferSplitShape (const ir::Shape input_shape, int axis_value, int num_splits)
ir::Shape	onert::shape_inference::inferSqueezeShape (const ir::Shape &in_shape, const ir::operation::Squeeze::Param &param)
template<typename T >
StridedSliceParams	onert::shape_inference::buildStridedSliceParams (const T *begin, const T *end, const T *strides, const uint32_t begin_mask, const uint32_t end_mask, const uint32_t shrink_axis_mask, const uint8_t rank)
ir::Shape	onert::shape_inference::inferStridedSliceShape (const ir::Shape &input_shape, const StridedSliceParams &op_params, uint32_t rank)
ir::Shape	onert::shape_inference::inferTileShape (const ir::Shape &in_shape, const int32_t *multiplier_buf, const int32_t multiplier_size)
ir::Shape	onert::shape_inference::inferTransposeShape (const ir::Shape &in_shape, const int32_t *perm_buf, const int32_t rank)
ir::Shape	onert::shape_inference::inferUnpackShape (const ir::Shape &input_shape, int axis, int rank)