| /frameworks/ml/nn/common/operations/ |
| D | GroupedConv2D.cpp | 39 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
72 for (uint32_t j = 0; j < filterWidth; j++) { in groupedConvFloat32()
80 i * filterWidth * filterDepth + j * filterDepth + k; in groupedConvFloat32()
91 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvFloat32()
144 for (uint32_t j = 0; j < filterWidth; j++) { in groupedConvQuant8()
152 i * filterWidth * filterDepth + j * filterDepth + k; in groupedConvQuant8()
169 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvQuant8()
248 for (uint32_t j = 0; j < filterWidth; j++) { in groupedConvQuant8PerChannel()
256 i * filterWidth * filterDepth + j * filterDepth + k; in groupedConvQuant8PerChannel()
273 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvQuant8PerChannel()
|
| D | TransposeConv2D.cpp | 76 int32_t filterWidth = getSizeOfDimension(filterShape, 2); in initialize() local
83 calculateExplicitPaddingTransposeConv(outputWidth, strideWidth, filterWidth, in initialize()
116 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
148 for (uint32_t j = 0; j < filterWidth; j++, filterBase += inputDepth) { in transposeConvNhwc()
233 for (uint32_t j = 0; j < filterWidth; j++) { in transposeConvNhwc()
240 k * filterHeight * filterWidth * inputDepth + in transposeConvNhwc()
241 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvNhwc()
375 for (uint32_t j = 0; j < filterWidth; j++) { in transposeConvQuant8PerChannelNhwc()
382 k * filterHeight * filterWidth * inputDepth + in transposeConvQuant8PerChannelNhwc()
383 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvQuant8PerChannelNhwc()
[all …]
|
| D | Conv2D.cpp | 136 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
148 im2colDim.sizes[0] = (int)inDepth * filterHeight * filterWidth; \
366 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); in convQuant8PerChannelNhwc() local
407 for (uint32_t j = 0; j < filterWidth; j++) { in convQuant8PerChannelNhwc()
417 i * filterWidth * filterDepth + j * filterDepth + k; in convQuant8PerChannelNhwc()
433 filterBase += filterHeight * filterWidth * filterDepth; in convQuant8PerChannelNhwc()
459 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); in convQuant8PerChannelNhwc() local
659 uint32_t filterWidth = getSizeOfDimension(filter, 2); in prepare() local
668 int32_t effectiveFilterWidth = (filterWidth - 1) * param.dilation_width_factor + 1; in prepare()
676 computeOutSize(width, filterWidth, param.stride_width, param.dilation_width_factor, in prepare()
|
| D | DepthwiseConv2D.cpp | 124 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
292 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); in depthwiseConvQuant8PerChannelNhwc() local
334 for (uint32_t j = 0; j < filterWidth; j++) { in depthwiseConvQuant8PerChannelNhwc()
343 i * filterWidth * filterDepth + j * filterDepth + oc; in depthwiseConvQuant8PerChannelNhwc()
543 uint32_t filterWidth = getSizeOfDimension(filter, 2); in prepare() local
546 int32_t effectiveFilterWidth = (filterWidth - 1) * param.dilation_width_factor + 1; in prepare()
557 computeOutSize(width, filterWidth, param.stride_width, param.dilation_width_factor, in prepare()
|
| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Poolings.cpp | 34 auto filterWidth = op->inputs[7]->value<RandomVariable>(); in poolingExplicitOpConstructor() local
59 explicitPadding(op->inputs[0]->dimensions[widthIndex], filterWidth, strideWidth, /*dilation=*/1, in poolingExplicitOpConstructor()
73 auto filterWidth = op->inputs[4]->value<RandomVariable>(); in poolingImplicitOpConstructor() local
95 implicitPadding(op->inputs[0]->dimensions[widthIndex], filterWidth, strideWidth, in poolingImplicitOpConstructor()
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 737 uint32_t filterWidth = getSizeOfDimension(filter, 2); in groupedConvPrepare() local
741 NN_RET_CHECK_GT(static_cast<int32_t>(filterWidth), padding_left); in groupedConvPrepare()
742 NN_RET_CHECK_GT(static_cast<int32_t>(filterWidth), padding_right); in groupedConvPrepare()
747 computeOutSize(width, filterWidth, stride_width, padding_left, padding_right); in groupedConvPrepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 238 int32_t strideHeight, int32_t filterWidth, in getPaddingScheme() argument
249 calculateExplicitPadding(inWidth, strideWidth, filterWidth, kPaddingSame, &expectedPaddingLeft, in getPaddingScheme()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1845 ANeuralNetworksOperandType filterWidth = scalar; in poolingOpTest() local
1851 strideHeight, filterWidth, filterHeight, activation}, in poolingOpTest()
1858 {input, padImplicit, strideWidth, strideHeight, filterWidth, filterHeight, activation}, in poolingOpTest()
1870 {input, padLeft, padRight, padTop, padBottom, strideWidth, strideHeight, filterWidth, in poolingOpTest()
1877 filterWidth, filterHeight, activation, layout}, in poolingOpTest()
|