Home
last modified time | relevance | path

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

123

/packages/modules/NeuralNetworks/common/cpu_operations/
DActivation.cpp73 const Shape& outputShape) { in relu1Float()
83 const Shape& outputShape) { in relu6Float()
144 const Shape& outputShape) { in reluQuant8()
150 const Shape& outputShape) { in relu1Quant8()
156 const Shape& outputShape) { in relu6Quant8()
162 const Shape& outputShape) { in tanhQuant8()
192 const Shape& outputShape) { in logisticQuant8()
239 const Shape& outputShape) { in reluQuant8Signed()
245 const Shape& outputShape) { in relu1Quant8Signed()
251 const Shape& outputShape) { in relu6Quant8Signed()
[all …]
DPooling.cpp144 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
154 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
167 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
177 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
190 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
200 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
212 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
222 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
232 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
244 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc()
[all …]
DReshape.cpp50 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
71 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric()
92 T* outputData, const Shape& outputShape) { in padGeneric()
205 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric()
231 const Shape& outputShape) { in spaceToBatchGeneric()
DFullyConnected.cpp55 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
85 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
106 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
147 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
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()
DQuantize.cpp36 bool quantizeToQuant8(const T* inputData, uint8_t* outputData, const Shape& outputShape) { in quantizeToQuant8()
49 bool quantizeToQuant8Signed(const T* inputData, int8_t* outputData, const Shape& outputShape) { in quantizeToQuant8Signed()
DSimpleMath.cpp41 const Shape& outputShape) { in meanFloat16()
55 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
DConcatenation.cpp54 const Shape& outputShape) { in concatenation()
74 uint8_t* outputData, const Shape& outputShape) { in concatenation()
133 Shape outputShape(context->getOutputShape(kOutputTensor)); in concatenation() local
DResizeImageOps.cpp62 bool halfPixelCenters, T* outputData, const Shape& outputShape) { in resizeNearestNeighbor()
106 const Shape& outputShape) { in resizeImageOpNhwc()
133 _Float16* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
147 const Shape& outputShape) { in resizeImageOp()
DCast.cpp45 const Shape& outputShape) { in copyToTensor()
73 const Shape& outputShape) { in eval()
DPow.cpp35 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
70 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
DRoiPooling.cpp45 T_Input* outputData, const Shape& outputShape) { in roiPoolingNhwc()
134 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
152 const Shape& outputShape) { in roiPooling()
167 const Shape& outputShape) { in roiPooling()
DL2Normalization.cpp138 const Shape& outputShape) { in l2normFloat32()
154 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
167 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
183 int8_t* outputData, const Shape& outputShape) { in l2normQuant8Signed()
DSlice.cpp47 T* outputData, const Shape& outputShape) { in evalGeneric()
93 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
DStridedSlice.cpp49 int32_t shrinkAxisMask, T* outputData, const Shape& outputShape) { in compute()
158 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
DConv2D.cpp196 float* outputData, const Shape& outputShape) { in convNhwc()
225 uint8_t* outputData, const Shape& outputShape) { in convNhwc()
279 int8_t* outputData, Shape outputShape) { in convNhwc()
308 _Float16* outputData, const Shape& outputShape) { in convNhwc()
335 const Shape& outputShape) { in conv()
356 const Shape& outputShape) { in convQuant8PerChannelNhwc()
449 const Shape& outputShape) { in convQuant8PerChannelNhwc()
512 T* outputData, const Shape& outputShape) { in convQuant8PerChannel()
DDepthwiseConv2D.cpp135 const Shape& outputShape) { in depthwiseConvNhwc()
169 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
195 const Shape& outputShape) { in depthwiseConvNhwc()
246 Shape outputShape) { in depthwiseConvNhwc()
279 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
375 const Shape& outputShape) { in depthwiseConv()
398 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel()
DTransposeConv2D.cpp125 const Shape& outputShape) { in transposeConvNhwc()
181 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
277 const Shape& outputShape) { in transposeConvNhwc()
300 const Shape& outputShape) { in transposeConv()
317 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
420 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
DPRelu.cpp46 const Shape& outputShape) { in eval()
69 T* outputData, const Shape& outputShape) { in evalQuant8()
DSoftmax.cpp83 float* outputData, const Shape& outputShape) { in softmaxFloat32()
99 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16()
201 T* outputData, const Shape& outputShape) { in softmaxQuant8()
DReverse.cpp56 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
DGroupedConv2D.cpp55 const Shape& outputShape) { in groupedConvFloat32()
113 const Shape& outputShape) { in groupedConvQuant8()
211 T* outputData, const Shape& outputShape) { in groupedConvQuant8PerChannel()
295 int32_t activation, _Float16* outputData, const Shape& outputShape) { in groupedConvFloat16()
/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp37 void CalculateActivationRangeImpl(int32_t activation, const Shape& outputShape, int32_t qmin, in CalculateActivationRangeImpl()
155 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultiplier()
170 const Shape& outputShape, double* multiplier) { in GetQuantizedConvolutionMultiplier()
180 void CalculateActivationRangeUint8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeUint8()
188 void CalculateActivationRangeInt8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeInt8()
369 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare()
388 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare()
/packages/modules/NeuralNetworks/common/types/operations/src/
DPack.cpp43 const Shape outputShape = context->getOutputShape(kOutputTensor); in validate() local
DCast.cpp32 auto outputShape = context->getOutputShape(0); in validate() local

123