Lines Matching refs:shape
105 virtual bool setOutputShape(uint32_t index, const Shape& shape) = 0;
147 uint32_t getNumberOfElements(const Shape& shape);
148 uint32_t getNumberOfElements(const Shape& shape, size_t firstAxisInclusive,
151 uint32_t getNumberOfDimensions(const Shape& shape);
153 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);
155 uint32_t hasKnownRank(const Shape& shape);
160 inline bool handleNegativeAxis(const Shape& shape, int32_t* axis) { in handleNegativeAxis() argument
161 return handleNegativeAxis(getNumberOfDimensions(shape), axis); in handleNegativeAxis()
342 inline bool transposeFirstTwoDimensions(const T* buffer, const Shape& shape, T* transposedBuffer) { in transposeFirstTwoDimensions() argument
343 const int numDims = getNumberOfDimensions(shape); in transposeFirstTwoDimensions()
345 const int firstDim = getSizeOfDimension(shape, 0); in transposeFirstTwoDimensions()
346 const int secondDim = getSizeOfDimension(shape, 1); in transposeFirstTwoDimensions()
349 blockSize *= getSizeOfDimension(shape, i); in transposeFirstTwoDimensions()
363 inline bool transposeFirstTwoDimensions(const Shape& shape, Shape* transposedShape) { in transposeFirstTwoDimensions() argument
364 NN_RET_CHECK(getNumberOfDimensions(shape) >= 2); in transposeFirstTwoDimensions()
365 *transposedShape = shape; in transposeFirstTwoDimensions()
366 transposedShape->dimensions[0] = shape.dimensions[1]; in transposeFirstTwoDimensions()
367 transposedShape->dimensions[1] = shape.dimensions[0]; in transposeFirstTwoDimensions()