Lines Matching refs:Shape
68 bool SameShape(const Shape& in1, const Shape& in2) { in SameShape()
80 bool SetShape(const Shape& in, Shape* out) { in SetShape()
115 uint32_t getNumberOfElements(const Shape& shape) { in getNumberOfElements()
123 uint32_t getNumberOfElements(const Shape& shape, in getNumberOfElements()
136 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions()
140 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension()
213 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()
214 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler()
229 const Shape& outputShape, in CalculateActivationRangeUint8()
312 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out) { in calculateBroadcastedShape()
338 uint8_t requantize(uint8_t value, const Shape& oldShape, const Shape& newShape) { in requantize()
346 bool floorPrepare(const Shape& input, Shape* output) { in floorPrepare()
350 bool depthwiseConvPrepare(const Shape& input, const Shape& filter, const Shape& bias, in depthwiseConvPrepare()
354 int32_t dilation_height_factor, Shape* output) { in depthwiseConvPrepare()
397 bool genericActivationPrepare(const Shape& input, in genericActivationPrepare()
398 Shape* output) { in genericActivationPrepare()
403 bool genericNormalizationPrepare(const Shape& input, Shape* output) { in genericNormalizationPrepare()
407 bool reshapePrepare(const Shape& input, in reshapePrepare()
410 Shape* output) { in reshapePrepare()
446 bool depthToSpacePrepare(const Shape& input, in depthToSpacePrepare()
448 Shape* output) { in depthToSpacePrepare()
469 bool spaceToDepthPrepare(const Shape& input, in spaceToDepthPrepare()
471 Shape* output) { in spaceToDepthPrepare()
494 bool embeddingLookupPrepare(const Shape &valueShape, in embeddingLookupPrepare()
495 const Shape &lookupShape, in embeddingLookupPrepare()
496 Shape *outputShape) { in embeddingLookupPrepare()
516 bool hashtableLookupPrepare(const Shape &lookupShape, in hashtableLookupPrepare()
517 const Shape &keyShape, in hashtableLookupPrepare()
518 const Shape &valueShape, in hashtableLookupPrepare()
519 Shape *outputShape, in hashtableLookupPrepare()
520 Shape *hitShape) { in hashtableLookupPrepare()
544 bool padPrepare(const Shape& input, in padPrepare()
546 const Shape& paddingsShape, in padPrepare()
547 Shape* output) { in padPrepare()
572 bool batchToSpacePrepare(const Shape& input, in batchToSpacePrepare()
574 const Shape& blockSizeShape, in batchToSpacePrepare()
575 Shape* output) { in batchToSpacePrepare()
602 bool spaceToBatchPrepare(const Shape& input, in spaceToBatchPrepare()
604 const Shape& blockSizeShape, in spaceToBatchPrepare()
606 const Shape& paddingsShape, in spaceToBatchPrepare()
607 Shape* output) { in spaceToBatchPrepare()
645 bool squeezePrepare(const Shape& input, in squeezePrepare()
647 const Shape& squeezeDimsShape, in squeezePrepare()
648 Shape* output) { in squeezePrepare()
694 bool meanPrepare(const Shape& input, in meanPrepare()
696 const Shape& axisShape, in meanPrepare()
698 Shape* output) { in meanPrepare()
771 bool stridedSlicePrepare(const Shape& input, in stridedSlicePrepare()
772 const int32_t* beginData, const Shape& beginShape, in stridedSlicePrepare()
773 const int32_t* endData, const Shape& endShape, in stridedSlicePrepare()
774 const int32_t* stridesData, const Shape& stridesShape, in stridedSlicePrepare()
776 Shape* output) { in stridedSlicePrepare()
830 bool argMinMaxPrepare(const Shape& input, int32_t axis, Shape* output) { in argMinMaxPrepare()
848 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, in splitPrepare()
849 std::vector<Shape>* output) { in splitPrepare()
866 bool groupedConvPrepare(const Shape& input, const Shape& filter, const Shape& bias, in groupedConvPrepare()
869 int32_t numGroups, Shape* output) { in groupedConvPrepare()