Home
last modified time | relevance | path

Searched defs:outputShape (Results 1 – 25 of 59) sorted by relevance

123

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DActivation.cpp71 const Shape& outputShape) { in relu1Float()
81 const Shape& outputShape) { in relu6Float()
90 const Shape& outputShape) { in tanhFloat16()
100 const Shape& outputShape) { in tanhFloat32()
111 const Shape& outputShape) { in logisticFloat()
142 const Shape& outputShape) { in reluQuant8()
148 const Shape& outputShape) { in relu1Quant8()
154 const Shape& outputShape) { in relu6Quant8()
160 const Shape& outputShape) { in tanhQuant8()
190 const Shape& outputShape) { in logisticQuant8()
[all …]
H A DPooling.cpp142 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
175 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
188 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
198 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
210 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
220 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
230 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
242 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc()
256 T* outputData, const Shape& outputShape) { in averagePool()
269 const Shape& outputShape) { in l2Pool()
[all …]
H A DReshape.cpp35 const Shape& outputShape) { in copyData()
44 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
65 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric()
86 T* outputData, const Shape& outputShape) { in padGeneric()
199 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric()
225 const Shape& outputShape) { in spaceToBatchGeneric()
H A DL2Normalization.cpp49 float* outputData, const Shape& outputShape) { in l2normFloat32Impl()
77 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8Impl()
109 int8_t* outputData, const Shape& outputShape) { in l2normQuant8SignedImpl()
140 const Shape& outputShape) { in l2normFloat32()
156 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
169 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
185 int8_t* outputData, const Shape& outputShape) { in l2normQuant8Signed()
H A DMaximumMinimum.cpp36 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
62 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8()
96 bool isMinimum, void* output, const Shape& outputShape) { in eval()
H A DSimpleMath.cpp35 const Shape& outputShape) { in meanFloat16()
49 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
H A DArgMinMax.cpp30 Out* outputData, const Shape& outputShape) { in argMinMaxImpl()
52 uint8_t* outputData, const Shape& outputShape) { in argMinMaxGeneric()
H A DCast.cpp45 const Shape& outputShape) { in copyToTensor()
73 const Shape& outputShape) { in eval()
H A DPow.cpp35 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
70 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
H A DQuantize.cpp40 bool quantizeToQuant8(const T* inputData, uint8_t* outputData, const Shape& outputShape) { in quantizeToQuant8()
52 bool quantizeToQuant8Signed(const T* inputData, int8_t* outputData, const Shape& outputShape) { in quantizeToQuant8Signed()
H A DSoftmax.cpp54 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32()
86 float* outputData, const Shape& outputShape) { in softmaxFloat32()
102 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16()
118 T* outputData, const Shape& outputShape) { in softmaxQuant8Impl()
204 T* outputData, const Shape& outputShape) { in softmaxQuant8()
H A DFullyConnected.cpp59 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
89 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
110 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
151 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
H A DTile.cpp70 const Shape& outputShape) { in tileImpl()
91 uint8_t* outputData, const Shape& outputShape) { in eval()
H A DExpandDims.cpp41 const Shape& outputShape) { in eval()
H A DSlice.cpp55 T* outputData, const Shape& outputShape) { in evalGeneric()
126 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
H A DResizeImageOps.cpp69 bool halfPixelCenters, T* outputData, const Shape& outputShape) { in resizeNearestNeighbor()
113 const Shape& outputShape) { in resizeImageOpNhwc()
154 const Shape& outputShape) { in resizeImageOp()
H A DConcatenation.cpp52 const Shape& outputShape) { in concatenation()
131 Shape outputShape(context->getOutputShape(kOutputTensor)); variable
H A DConv2D.cpp199 float* outputData, const Shape& outputShape) { in convNhwc()
228 uint8_t* outputData, const Shape& outputShape) { in convNhwc()
282 int8_t* outputData, Shape outputShape) { in convNhwc()
311 _Float16* outputData, const Shape& outputShape) { in convNhwc()
338 const Shape& outputShape) { in conv()
359 const Shape& outputShape) { in convQuant8PerChannelNhwc()
452 const Shape& outputShape) { in convQuant8PerChannelNhwc()
515 T* outputData, const Shape& outputShape) { in convQuant8PerChannel()
H A DInstanceNormalization.cpp50 T epsilon, T* outputData, const Shape& outputShape) { in instanceNormNhwc()
93 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
H A DTransposeConv2D.cpp129 const Shape& outputShape) { in transposeConvNhwc()
185 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
281 const Shape& outputShape) { in transposeConvNhwc()
304 const Shape& outputShape) { in transposeConv()
321 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
424 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
H A DDepthwiseConv2D.cpp139 const Shape& outputShape) { in depthwiseConvNhwc()
173 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
199 const Shape& outputShape) { in depthwiseConvNhwc()
250 Shape outputShape) { in depthwiseConvNhwc()
283 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
379 const Shape& outputShape) { in depthwiseConv()
402 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel()
H A DGroupedConv2D.cpp51 const Shape& outputShape) { in groupedConvFloat32()
109 const Shape& outputShape) { in groupedConvQuant8()
206 int32_t activation, T* outputData, const Shape& outputShape) { in groupedConvQuant8PerChannel()
290 int32_t activation, _Float16* outputData, const Shape& outputShape) { in groupedConvFloat16()
H A DPRelu.cpp48 const Shape& outputShape) { in eval()
71 T* outputData, const Shape& outputShape) { in evalQuant8()
H A DStridedSlice.cpp55 int32_t shrinkAxisMask, T* outputData, const Shape& outputShape) { in compute()
202 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
/aosp12/packages/modules/NeuralNetworks/common/
H A DOperationsUtils.cpp48 void CalculateActivationRangeImpl(int32_t activation, const Shape& outputShape, int32_t qmin, in CalculateActivationRangeImpl()
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()
460 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare()
479 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare()

123