Home
last modified time | relevance | path

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

123

/packages/modules/NeuralNetworks/common/operations/
DActivation.cpp53 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
71 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()
126 const Shape& outputShape) { in reluXQuant8()
142 const Shape& outputShape) { in reluQuant8()
148 const Shape& outputShape) { in relu1Quant8()
154 const Shape& outputShape) { in relu6Quant8()
[all …]
DPooling.cpp142 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
152 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
165 uint8_t* 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()
[all …]
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()
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()
DSimpleMath.cpp35 const Shape& outputShape) { in meanFloat16()
49 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
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()
DArgMinMax.cpp30 Out* outputData, const Shape& outputShape) { in argMinMaxImpl()
52 uint8_t* outputData, const Shape& outputShape) { in argMinMaxGeneric()
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()
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()
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()
DConcatenation.cpp52 const Shape& outputShape) { in concatenation()
72 uint8_t* outputData, const Shape& outputShape) { in concatenation()
131 Shape outputShape(context->getOutputShape(kOutputTensor)); in concatenation() local
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()
DResizeImageOps.cpp69 bool halfPixelCenters, T* outputData, const Shape& outputShape) { in resizeNearestNeighbor()
113 const Shape& outputShape) { in resizeImageOpNhwc()
140 _Float16* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
154 const Shape& outputShape) { in resizeImageOp()
DExpandDims.cpp41 const Shape& outputShape) { in eval()
DLocalResponseNormalization.cpp54 const Shape& outputShape) { in localResponseNormFloat32Impl()
88 const Shape& outputShape) { in localResponseNorm()
110 _Float16* outputData, const Shape& outputShape) { in localResponseNorm()
DTile.cpp70 const Shape& outputShape) { in tileImpl()
91 uint8_t* outputData, const Shape& outputShape) { in eval()
DRoiPooling.cpp58 T_Input* outputData, const Shape& outputShape) { in roiPoolingNhwc()
147 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
165 const Shape& outputShape) { in roiPooling()
180 const Shape& outputShape) { in roiPooling()
DSlice.cpp55 T* outputData, const Shape& outputShape) { in evalGeneric()
126 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
DInstanceNormalization.cpp50 T epsilon, T* outputData, const Shape& outputShape) { in instanceNormNhwc()
93 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
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()
DPRelu.cpp48 const Shape& outputShape) { in eval()
71 T* outputData, const Shape& outputShape) { in evalQuant8()
DStridedSlice.cpp55 int32_t shrinkAxisMask, T* outputData, const Shape& outputShape) { in compute()
202 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
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()
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()

123