luci::sinf::TensorShapeExpander Class Reference

Create a higher-rank TensorShape following NumPy broadcasting semantics. More...

#include <CircleShapeInferenceHelper.h>

Public Member Functions
	TensorShapeExpander (const loco::TensorShape &shape)
loco::TensorShape	to (uint32_t output_rank)

Detailed Description

Create a higher-rank TensorShape following NumPy broadcasting semantics.

HOW TO USE:

auto expanded_tensor_shape = expand(tensor_shape).to(N);

Definition at line 62 of file CircleShapeInferenceHelper.h.

Constructor & Destructor Documentation

TensorShapeExpander()

luci::sinf::TensorShapeExpander::TensorShapeExpander ( const loco::TensorShape &  shape )

inline

Definition at line 65 of file CircleShapeInferenceHelper.h.

                                                    : _shape{shape}
  {
    // DO NOTHING
  }

Member Function Documentation

to()

loco::TensorShape luci::sinf::TensorShapeExpander::to ( uint32_t  output_rank )

inline

Definition at line 71 of file CircleShapeInferenceHelper.h.

  {
    auto const &input_shape = _shape;
    uint32_t const input_rank = input_shape.rank();
 
    assert(input_rank <= output_rank && "Cannot shrink rank");
    uint32_t const axis_shift = output_rank - input_rank;
 
    loco::TensorShape output_shape;
 
    output_shape.rank(output_rank);
    for (uint32_t axis = 0; axis < output_rank; ++axis)
    {
      output_shape.dim(axis) = (axis < axis_shift) ? 1 : input_shape.dim(axis - axis_shift);
    }
 
    return output_shape;
  }

References luci::must_cast(), output_shape, and loco::TensorShape::rank().

---

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

• compiler/luci/service/src/CircleShapeInferenceHelper.h