Lines Matching refs:Shape
45 struct Shape { struct
76 virtual Shape getInputShape(uint32_t index) const = 0;
81 virtual Shape getOutputShape(uint32_t index) const = 0;
91 virtual Shape getInputShape(uint32_t index) const = 0;
97 virtual Shape getOutputShape(uint32_t index) const = 0;
101 virtual bool setOutputShape(uint32_t index, const Shape& shape) = 0;
136 bool SameShape(const Shape& in1, const Shape& in2);
139 bool SetShape(const Shape& in, Shape* out);
147 uint32_t getNumberOfElements(const Shape& shape);
148 uint32_t getNumberOfElements(const Shape& shape,
152 uint32_t getNumberOfDimensions(const Shape& shape);
154 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);
159 inline bool handleNegativeAxis(const Shape& shape, int32_t* axis) { in handleNegativeAxis()
191 __wur bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape,
192 const Shape& biasShape, const Shape& outputShape,
196 const Shape& outputShape,
288 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out);
291 uint8_t requantize(uint8_t value, const Shape& oldShape, const Shape& newShape);
294 bool floorPrepare(const Shape& input, Shape* output);
296 bool depthwiseConvPrepare(const Shape& input, const Shape& filter, const Shape& bias,
300 int32_t dilation_height_factor, Shape* output);
302 bool genericActivationPrepare(const Shape& input, Shape* output);
304 bool genericNormalizationPrepare(const Shape& input, Shape* output);
306 bool reshapePrepare(const Shape& input,
309 Shape* output);
311 bool depthToSpacePrepare(const Shape& input,
313 Shape* output);
315 bool spaceToDepthPrepare(const Shape& input,
317 Shape* output);
319 bool embeddingLookupPrepare(const Shape &valueShape,
320 const Shape &lookupShape,
321 Shape *outputShape);
323 bool hashtableLookupPrepare(const Shape &lookupShape,
324 const Shape &keyShape,
325 const Shape &valueShape,
326 Shape *outputShape,
327 Shape *hitShape);
329 bool padPrepare(const Shape& input,
331 const Shape& paddingsShape,
332 Shape* output);
334 bool batchToSpacePrepare(const Shape& input,
336 const Shape& blockSizeShape,
337 Shape* output);
339 bool spaceToBatchPrepare(const Shape& input,
341 const Shape& blockSizeShape,
343 const Shape& paddingsShape,
344 Shape* output);
346 bool squeezePrepare(const Shape& input,
348 const Shape& squeezeDimsShape,
349 Shape* output);
351 bool meanPrepare(const Shape& input,
353 const Shape& axisShape,
355 Shape* output);
357 bool stridedSlicePrepare(const Shape& input,
358 const int32_t* beginData, const Shape& beginShape,
359 const int32_t* endData, const Shape& endShape,
360 const int32_t* stridesData, const Shape& stridesShape,
362 Shape* output);
364 bool argMinMaxPrepare(const Shape& input, int32_t axis, Shape* output);
366 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, std::vector<Shape>* output);
368 bool groupedConvPrepare(const Shape& input, const Shape& filter, const Shape& bias,
371 int32_t numGroups, Shape* output);
375 inline bool transposeFirstTwoDimensions(const T* buffer, const Shape& shape, T* transposedBuffer) { in transposeFirstTwoDimensions()
396 inline bool transposeFirstTwoDimensions(const Shape& shape, Shape* transposedShape) { in transposeFirstTwoDimensions()