Lines Matching refs:Shape
48 void CalculateActivationRangeImpl(int32_t activation, const Shape& outputShape, int32_t qmin, in CalculateActivationRangeImpl()
114 bool SameShape(const Shape& in1, const Shape& in2) { in SameShape()
126 bool SetShape(const Shape& in, Shape* out) { in SetShape()
134 uint32_t getNumberOfElements(const Shape& shape) { in getNumberOfElements()
142 uint32_t getNumberOfElements(const Shape& shape, size_t firstAxisInclusive, in getNumberOfElements()
154 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions()
158 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension()
163 uint32_t hasKnownRank(const Shape& shape) { in hasKnownRank()
256 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()
257 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler()
271 void CalculateActivationRangeUint8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeUint8()
279 void CalculateActivationRangeInt8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeInt8()
339 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out) { in calculateBroadcastedShape()
366 uint8_t requantize<uint8_t>(uint8_t value, const Shape& oldShape, const Shape& newShape) { in requantize()
375 int8_t requantize<int8_t>(int8_t value, const Shape& oldShape, const Shape& newShape) { in requantize()
383 bool reshapePrepare(const Shape& input, const int32_t* targetDims, const int32_t targetDimsSize, in reshapePrepare()
384 Shape* output) { in reshapePrepare()
420 bool depthToSpacePrepare(const Shape& input, int32_t blockSize, Shape* output) { in depthToSpacePrepare()
439 bool spaceToDepthPrepare(const Shape& input, int32_t blockSize, Shape* output) { in spaceToDepthPrepare()
460 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare()
478 bool hashtableLookupPrepare(const Shape& lookupShape, const Shape& keyShape, in hashtableLookupPrepare()
479 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare()
501 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape, in padPrepare()
502 Shape* output) { in padPrepare()
527 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData, in batchToSpacePrepare()
528 const Shape& blockSizeShape, Shape* output) { in batchToSpacePrepare()
553 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData, in spaceToBatchPrepare()
554 const Shape& blockSizeShape, const int32_t* paddingsData, in spaceToBatchPrepare()
555 const Shape& paddingsShape, Shape* output) { in spaceToBatchPrepare()
592 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()
593 Shape* output) { in meanPrepare()
669 bool argMinMaxPrepare(const Shape& input, int32_t axis, Shape* output) { in argMinMaxPrepare()
689 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, in splitPrepare()
690 std::vector<Shape>* output) { in splitPrepare()
707 bool groupedConvPrepare(const Shape& input, const Shape& filter, const Shape& bias, in groupedConvPrepare()
710 int32_t numGroups, Shape* output) { in groupedConvPrepare()