Lines Matching refs:shape
34 inline tflite::Dims<4> convertShapeToDims(const Shape& shape) { in convertShapeToDims() argument
35 nnAssert(shape.dimensions.size() <= 4); in convertShapeToDims()
40 int src = static_cast<int>(shape.dimensions.size()) - i - 1; in convertShapeToDims()
42 dims.sizes[i] = static_cast<int>(getSizeOfDimension(shape, src)); in convertShapeToDims()
55 inline tflite::RuntimeShape convertShapeToTflshape(const Shape& shape) { in convertShapeToTflshape() argument
56 std::vector<int32_t> tflShapeDim(shape.dimensions.begin(), shape.dimensions.end()); in convertShapeToTflshape()
161 bool initialize(const T* data, const Shape& shape) { in initialize() argument
163 mShape = shape; in initialize()
165 return convertNchwToNhwc(mDataOriginal, shape, &mDataNhwc, &mShape); in initialize()
185 bool initialize(T* data, const Shape& shape) { in initialize() argument
186 NN_RET_CHECK_EQ(getNumberOfDimensions(shape), 4); in initialize()
188 mShape = shape; in initialize()
190 const auto& dim = shape.dimensions; in initialize()
192 mDataNhwc.resize(getNumberOfElements(shape)); in initialize()