Lines Matching refs:shape
35 inline tflite::Dims<4> convertShapeToDims(const Shape& shape) { in convertShapeToDims() argument
36 CHECK_LE(shape.dimensions.size(), 4u); in convertShapeToDims()
41 int src = static_cast<int>(shape.dimensions.size()) - i - 1; in convertShapeToDims()
43 dims.sizes[i] = static_cast<int>(getSizeOfDimension(shape, src)); in convertShapeToDims()
56 inline tflite::RuntimeShape convertShapeToTflshape(const Shape& shape) { in convertShapeToTflshape() argument
57 std::vector<int32_t> tflShapeDim(shape.dimensions.begin(), shape.dimensions.end()); in convertShapeToTflshape()
162 bool initialize(const T* data, const Shape& shape) { in initialize() argument
164 mShape = shape; in initialize()
166 return convertNchwToNhwc(mDataOriginal, shape, &mDataNhwc, &mShape); in initialize()
186 bool initialize(T* data, const Shape& shape) { in initialize() argument
187 NN_RET_CHECK_EQ(getNumberOfDimensions(shape), 4); in initialize()
189 mShape = shape; in initialize()
191 const auto& dim = shape.dimensions; in initialize()
193 mDataNhwc.resize(getNumberOfElements(shape)); in initialize()