Home
last modified time | relevance | path

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

123

/frameworks/ml/nn/common/operations/
DActivation.cpp52 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
70 const Shape& outputShape) { in relu1Float()
80 const Shape& outputShape) { in relu6Float()
89 const Shape& outputShape) { in tanhFloat16()
99 const Shape& outputShape) { in tanhFloat32()
110 const Shape& outputShape) { in logisticFloat()
125 const Shape& outputShape) { in reluXQuant8()
141 const Shape& outputShape) { in reluQuant8()
147 const Shape& outputShape) { in relu1Quant8()
153 const Shape& outputShape) { in relu6Quant8()
[all …]
DPooling.cpp140 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
150 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
163 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
173 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
186 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
196 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
208 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
218 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
228 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
240 _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.cpp48 float* outputData, const Shape& outputShape) { in l2normFloat32Impl()
76 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8Impl()
108 int8_t* outputData, const Shape& outputShape) { in l2normQuant8SignedImpl()
139 const Shape& outputShape) { in l2normFloat32()
155 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
168 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
184 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.cpp38 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
64 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8()
98 bool isMinimum, void* output, const Shape& outputShape) { in eval()
DArgMinMax.cpp34 Out* outputData, const Shape& outputShape) { in argMinMaxImpl()
56 uint8_t* outputData, const Shape& outputShape) { in argMinMaxGeneric()
DCast.cpp48 const Shape& outputShape) { in copyToTensor()
76 const Shape& outputShape) { in eval()
DPow.cpp36 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
71 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
DQuantize.cpp43 bool quantizeToQuant8(const T* inputData, uint8_t* outputData, const Shape& outputShape) { in quantizeToQuant8()
55 bool quantizeToQuant8Signed(const T* inputData, int8_t* outputData, const Shape& outputShape) { in quantizeToQuant8Signed()
DSoftmax.cpp51 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32()
83 float* outputData, const Shape& outputShape) { in softmaxFloat32()
99 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16()
115 T* outputData, const Shape& outputShape) { in softmaxQuant8Impl()
201 T* outputData, const Shape& outputShape) { in softmaxQuant8()
DConcatenation.cpp50 const Shape& outputShape) { in concatenation()
70 uint8_t* outputData, const Shape& outputShape) { in concatenation()
129 Shape outputShape(context->getOutputShape(kOutputTensor)); in concatenation() local
DFullyConnected.cpp58 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
88 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
109 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
150 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
DLocalResponseNormalization.cpp53 const Shape& outputShape) { in localResponseNormFloat32Impl()
87 const Shape& outputShape) { in localResponseNorm()
108 _Float16* outputData, const Shape& outputShape) { in localResponseNorm()
DResizeImageOps.cpp67 bool halfPixelCenters, T* outputData, const Shape& outputShape) { in resizeNearestNeighbor()
111 const Shape& outputShape) { in resizeImageOpNhwc()
138 _Float16* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
152 const Shape& outputShape) { in resizeImageOp()
DTile.cpp72 const Shape& outputShape) { in tileImpl()
93 uint8_t* outputData, const Shape& outputShape) { in eval()
DRoiPooling.cpp57 T_Input* outputData, const Shape& outputShape) { in roiPoolingNhwc()
146 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
164 const Shape& outputShape) { in roiPooling()
179 const Shape& outputShape) { in roiPooling()
DExpandDims.cpp41 const Shape& outputShape) { in eval()
DSlice.cpp54 T* outputData, const Shape& outputShape) { in evalGeneric()
121 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
DInstanceNormalization.cpp49 T epsilon, T* outputData, const Shape& outputShape) { in instanceNormNhwc()
92 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
DConv2D.cpp182 float* outputData, const Shape& outputShape) { in convNhwc()
207 uint8_t* outputData, const Shape& outputShape) { in convNhwc()
256 int8_t* outputData, Shape outputShape) { in convNhwc()
285 _Float16* outputData, const Shape& outputShape) { in convNhwc()
312 const Shape& outputShape) { in conv()
333 const Shape& outputShape) { in convQuant8PerChannelNhwc()
426 const Shape& outputShape) { in convQuant8PerChannelNhwc()
489 T* outputData, const Shape& outputShape) { in convQuant8PerChannel()
DTransposeConv2D.cpp128 const Shape& outputShape) { in transposeConvNhwc()
184 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
280 const Shape& outputShape) { in transposeConvNhwc()
303 const Shape& outputShape) { in transposeConv()
320 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
423 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
DStridedSlice.cpp54 int32_t shrinkAxisMask, T* outputData, const Shape& outputShape) { in compute()
199 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
DPRelu.cpp48 const Shape& outputShape) { in eval()
71 T* outputData, const Shape& outputShape) { in evalQuant8()
DDepthwiseConv2D.cpp137 const Shape& outputShape) { in depthwiseConvNhwc()
171 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
197 const Shape& outputShape) { in depthwiseConvNhwc()
248 Shape outputShape) { in depthwiseConvNhwc()
281 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
377 const Shape& outputShape) { in depthwiseConv()
400 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel()

123