Lines Matching refs:Shape

47 struct Shape {  struct
64     virtual Shape getInputShape(uint32_t index) const = 0;  argument
69     virtual Shape getOutputShape(uint32_t index) const = 0;
79     virtual Shape getInputShape(uint32_t index) const = 0;
85     virtual Shape getOutputShape(uint32_t index) const = 0;
89     virtual bool setOutputShape(uint32_t index, const Shape& shape) = 0;
124 bool SameShape(const Shape& in1, const Shape& in2);
127 bool SetShape(const Shape& in, Shape* out);
131 uint32_t getNumberOfElements(const Shape& shape);
132 uint32_t getNumberOfElements(const Shape& shape, size_t firstAxisInclusive,
135 uint32_t getNumberOfDimensions(const Shape& shape);
137 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);
139 uint32_t hasKnownRank(const Shape& shape);
144 inline bool handleNegativeAxis(const Shape& shape, int32_t* axis) {  in handleNegativeAxis()
179 __wur bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape,
180                                             const Shape& biasShape, const Shape& outputShape,
183 void CalculateActivationRangeUint8(int32_t activation, const Shape& outputShape, int32_t* act_min,
186 void CalculateActivationRangeInt8(int32_t activation, const Shape& outputShape, int32_t* act_min,
273 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out);
277 T requantize(T value, const Shape& oldShape, const Shape& newShape);
280 bool floorPrepare(const Shape& input, Shape* output);
282 bool depthwiseConvPrepare(const Shape& input, const Shape& filter, const Shape& bias,
286                           int32_t dilation_height_factor, Shape* output);
288 bool genericActivationPrepare(const Shape& input, Shape* output);
290 bool reshapePrepare(const Shape& input, const int32_t* targetDims, const int32_t targetDimsSize,
291                     Shape* output);
293 bool depthToSpacePrepare(const Shape& input, int32_t blockSize, Shape* output);
295 bool spaceToDepthPrepare(const Shape& input, int32_t blockSize, Shape* output);
297 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape);
299 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape,
300                             const Shape& valueShape, Shape* outputShape, Shape* hitShape);
302 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape,
303                 Shape* output);
305 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData,
306                          const Shape& blockSizeShape, Shape* output);
308 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData,
309                          const Shape& blockSizeShape, const int32_t* paddingsData,
310                          const Shape& paddingsShape, Shape* output);
312 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims,
313                  Shape* output);
315 bool argMinMaxPrepare(const Shape& input, int32_t axis, Shape* output);
317 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, std::vector<Shape>* output);
319 bool groupedConvPrepare(const Shape& input, const Shape& filter, const Shape& bias,
322                         int32_t numGroups, Shape* output);
326 inline bool transposeFirstTwoDimensions(const T* buffer, const Shape& shape, T* transposedBuffer) {  in transposeFirstTwoDimensions()
347 inline bool transposeFirstTwoDimensions(const Shape& shape, Shape* transposedShape) {  in transposeFirstTwoDimensions()