Searched refs:getNhwcShape (Results 1 – 9 of 9) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Pooling.cpp | 216 NN_RET_CHECK(averagePoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in averagePool()
217 output.getNhwcBuffer(), output.getNhwcShape())); in averagePool()
229 NN_RET_CHECK(l2PoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in l2Pool()
230 output.getNhwcBuffer(), output.getNhwcShape())); in l2Pool()
242 NN_RET_CHECK(maxPoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in maxPool()
243 output.getNhwcBuffer(), output.getNhwcShape())); in maxPool()
|
| D | InstanceNormalization.cpp | 84 NN_RET_CHECK(instanceNormNhwc(input.getNhwcBuffer(), input.getNhwcShape(), gamma, beta, epsilon, in instanceNorm()
85 output.getNhwcBuffer(), output.getNhwcShape())); in instanceNorm()
|
| D | ResizeImageOps.cpp | 95 NN_RET_CHECK(resizeImageOpNhwc(opType, input.getNhwcBuffer(), input.getNhwcShape(), in resizeImageOp()
96 output.getNhwcBuffer(), output.getNhwcShape())); in resizeImageOp()
|
| D | TransposeConv2D.cpp | 298 NN_RET_CHECK(transposeConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in transposeConv()
300 output.getNhwcShape())); in transposeConv()
417 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in transposeConvQuant8PerChannel()
418 biasData, biasShape, param, output.getNhwcBuffer(), output.getNhwcShape())); in transposeConvQuant8PerChannel()
|
| D | Conv2D.cpp | 273 NN_RET_CHECK(convNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, in conv()
277 output.getNhwcShape())); in conv()
388 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in convQuant8PerChannel()
391 output.getNhwcBuffer(), output.getNhwcShape())); in convQuant8PerChannel()
|
| D | RoiPooling.cpp | 145 NN_RET_CHECK(roiPoolingNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiPooling()
147 output.getNhwcBuffer(), output.getNhwcShape())); in roiPooling()
|
| D | RoiAlign.cpp | 320 NN_RET_CHECK(roiAlignNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiAlign()
323 output.getNhwcShape())); in roiAlign()
|
| D | GenerateProposals.cpp | 958 score_nhwc.getNhwcBuffer(), score_nhwc.getNhwcShape(), delta_nhwc.getNhwcBuffer(), in generateProposalsFloat32Compute()
959 delta_nhwc.getNhwcShape(), anchorsData, anchorsShape, imageInfoData, imageInfoShape, in generateProposalsFloat32Compute()
|
| /frameworks/ml/nn/common/include/ |
| D | CpuOperationUtils.h | 153 const Shape& getNhwcShape() { return mShape; } in getNhwcShape() function
180 const Shape& getNhwcShape() { return mShape; } in getNhwcShape() function
|