Searched refs:getNhwcShape (Results 1 – 10 of 10) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Pooling.cpp | 263 NN_RET_CHECK(averagePoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in averagePool()
264 output.getNhwcBuffer(), output.getNhwcShape())); in averagePool()
276 NN_RET_CHECK(l2PoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in l2Pool()
277 output.getNhwcBuffer(), output.getNhwcShape())); in l2Pool()
289 NN_RET_CHECK(maxPoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in maxPool()
290 output.getNhwcBuffer(), output.getNhwcShape())); in maxPool()
|
| D | InstanceNormalization.cpp | 88 NN_RET_CHECK(instanceNormNhwc(input.getNhwcBuffer(), input.getNhwcShape(), gamma, beta, epsilon, in instanceNorm()
89 output.getNhwcBuffer(), output.getNhwcShape())); in instanceNorm()
|
| D | RoiAlign.cpp | 317 input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, batchSplitData, in roiAlign()
319 output.getNhwcBuffer(), output.getNhwcShape())); in roiAlign()
321 NN_RET_CHECK(roiAlignNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiAlign()
324 output.getNhwcShape())); in roiAlign()
|
| D | TransposeConv2D.cpp | 305 NN_RET_CHECK(transposeConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in transposeConv()
307 output.getNhwcShape())); in transposeConv()
426 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in transposeConvQuant8PerChannel()
427 biasData, biasShape, param, output.getNhwcBuffer(), output.getNhwcShape())); in transposeConvQuant8PerChannel()
|
| D | ResizeImageOps.cpp | 152 NN_RET_CHECK(resizeImageOpNhwc(opType, input.getNhwcBuffer(), input.getNhwcShape(), in resizeImageOp()
154 output.getNhwcShape())); in resizeImageOp()
|
| D | DepthwiseConv2D.cpp | 380 NN_RET_CHECK(depthwiseConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in depthwiseConv()
384 activation, output.getNhwcBuffer(), output.getNhwcShape())); in depthwiseConv()
404 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in depthwiseConvQuant8PerChannel()
407 output.getNhwcBuffer(), output.getNhwcShape())); in depthwiseConvQuant8PerChannel()
|
| D | RoiPooling.cpp | 139 NN_RET_CHECK(roiPoolingNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiPooling()
141 output.getNhwcBuffer(), output.getNhwcShape())); in roiPooling()
|
| D | Conv2D.cpp | 340 NN_RET_CHECK(convNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, in conv()
344 output.getNhwcShape())); in conv()
518 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in convQuant8PerChannel()
521 output.getNhwcBuffer(), output.getNhwcShape())); in convQuant8PerChannel()
|
| D | GenerateProposals.cpp | 961 score_nhwc.getNhwcBuffer(), score_nhwc.getNhwcShape(), delta_nhwc.getNhwcBuffer(), in generateProposalsFloat32Compute()
962 delta_nhwc.getNhwcShape(), anchorsData, anchorsShape, imageInfoData, imageInfoShape, in generateProposalsFloat32Compute()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | CpuOperationUtils.h | 172 const Shape& getNhwcShape() { return mShape; } in getNhwcShape() function
199 const Shape& getNhwcShape() { return mShape; } in getNhwcShape() function
|