Searched refs:getNhwcShape (Results 1 – 10 of 10) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | Pooling.cpp | 261 NN_RET_CHECK(averagePoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in averagePool()
262 output.getNhwcBuffer(), output.getNhwcShape())); in averagePool()
274 NN_RET_CHECK(l2PoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in l2Pool()
275 output.getNhwcBuffer(), output.getNhwcShape())); in l2Pool()
287 NN_RET_CHECK(maxPoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in maxPool()
288 output.getNhwcBuffer(), output.getNhwcShape())); in maxPool()
|
| D | InstanceNormalization.cpp | 98 NN_RET_CHECK(instanceNormNhwc(input.getNhwcBuffer(), input.getNhwcShape(), gamma, beta, epsilon, in instanceNorm()
99 output.getNhwcBuffer(), output.getNhwcShape())); in instanceNorm()
|
| D | RoiAlign.cpp | 329 input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, batchSplitData, in roiAlign()
331 output.getNhwcBuffer(), output.getNhwcShape())); in roiAlign()
333 NN_RET_CHECK(roiAlignNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiAlign()
336 output.getNhwcShape())); in roiAlign()
|
| D | TransposeConv2D.cpp | 309 NN_RET_CHECK(transposeConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in transposeConv()
311 output.getNhwcShape())); in transposeConv()
430 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in transposeConvQuant8PerChannel()
431 biasData, biasShape, param, output.getNhwcBuffer(), output.getNhwcShape())); in transposeConvQuant8PerChannel()
|
| D | DepthwiseConv2D.cpp | 384 NN_RET_CHECK(depthwiseConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in depthwiseConv()
388 activation, output.getNhwcBuffer(), output.getNhwcShape())); in depthwiseConv()
408 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in depthwiseConvQuant8PerChannel()
411 output.getNhwcBuffer(), output.getNhwcShape())); in depthwiseConvQuant8PerChannel()
|
| D | Conv2D.cpp | 343 NN_RET_CHECK(convNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, in conv()
347 output.getNhwcShape())); in conv()
521 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in convQuant8PerChannel()
524 output.getNhwcBuffer(), output.getNhwcShape())); in convQuant8PerChannel()
|
| D | ResizeImageOps.cpp | 159 NN_RET_CHECK(resizeImageOpNhwc(opType, input.getNhwcBuffer(), input.getNhwcShape(), in resizeImageOp()
161 output.getNhwcShape())); in resizeImageOp()
|
| D | RoiPooling.cpp | 152 NN_RET_CHECK(roiPoolingNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiPooling()
154 output.getNhwcBuffer(), output.getNhwcShape())); in roiPooling()
|
| D | GenerateProposals.cpp | 1074 score_nhwc.getNhwcBuffer(), score_nhwc.getNhwcShape(), delta_nhwc.getNhwcBuffer(), in generateProposalsFloat32Compute()
1075 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
|