| /frameworks/ml/nn/common/operations/ |
| D | TransposeConv2D.cpp | 77 int32_t filterHeight = getSizeOfDimension(filterShape, 1); in initialize() local
85 calculateExplicitPaddingTransposeConv(outputHeight, strideHeight, filterHeight, in initialize()
115 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
147 for (uint32_t i = 0; i < filterHeight; i++) { in transposeConvNhwc()
232 for (uint32_t i = 0; i < filterHeight; i++) { in transposeConvNhwc()
240 k * filterHeight * filterWidth * inputDepth + in transposeConvNhwc()
374 for (uint32_t i = 0; i < filterHeight; i++) { in transposeConvQuant8PerChannelNhwc()
382 k * filterHeight * filterWidth * inputDepth + in transposeConvQuant8PerChannelNhwc()
512 uint32_t filterHeight = getSizeOfDimension(filter, 1); in prepare() local
522 NN_RET_CHECK_GT(filterHeight, 0); in prepare()
[all …]
|
| D | GroupedConv2D.cpp | 38 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
71 for (uint32_t i = 0; i < filterHeight; i++) { in groupedConvFloat32()
91 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvFloat32()
143 for (uint32_t i = 0; i < filterHeight; i++) { in groupedConvQuant8()
169 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvQuant8()
247 for (uint32_t i = 0; i < filterHeight; i++) { in groupedConvQuant8PerChannel()
273 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvQuant8PerChannel()
|
| D | Conv2D.cpp | 135 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
148 im2colDim.sizes[0] = (int)inDepth * filterHeight * filterWidth; \
365 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); in convQuant8PerChannelNhwc() local
406 for (uint32_t i = 0; i < filterHeight; i++) { in convQuant8PerChannelNhwc()
433 filterBase += filterHeight * filterWidth * filterDepth; in convQuant8PerChannelNhwc()
458 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); in convQuant8PerChannelNhwc() local
658 uint32_t filterHeight = getSizeOfDimension(filter, 1); in prepare() local
669 int32_t effectiveFilterHeight = (filterHeight - 1) * param.dilation_height_factor + 1; in prepare()
679 computeOutSize(height, filterHeight, param.stride_height, param.dilation_height_factor, in prepare()
|
| D | DepthwiseConv2D.cpp | 123 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
291 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); in depthwiseConvQuant8PerChannelNhwc() local
333 for (uint32_t i = 0; i < filterHeight; i++) { in depthwiseConvQuant8PerChannelNhwc()
542 uint32_t filterHeight = getSizeOfDimension(filter, 1); in prepare() local
547 int32_t effectiveFilterHeight = (filterHeight - 1) * param.dilation_height_factor + 1; in prepare()
554 computeOutSize(height, filterHeight, param.stride_height, param.dilation_height_factor, in prepare()
|
| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Poolings.cpp | 35 auto filterHeight = op->inputs[8]->value<RandomVariable>(); in poolingExplicitOpConstructor() local
54 explicitPadding(op->inputs[0]->dimensions[heightIndex], filterHeight, strideHeight, in poolingExplicitOpConstructor()
74 auto filterHeight = op->inputs[5]->value<RandomVariable>(); in poolingImplicitOpConstructor() local
93 implicitPadding(op->inputs[0]->dimensions[heightIndex], filterHeight, strideHeight, in poolingImplicitOpConstructor()
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 738 uint32_t filterHeight = getSizeOfDimension(filter, 1); in groupedConvPrepare() local
743 NN_RET_CHECK_GT(static_cast<int32_t>(filterHeight), padding_top); in groupedConvPrepare()
744 NN_RET_CHECK_GT(static_cast<int32_t>(filterHeight), padding_bottom); in groupedConvPrepare()
749 computeOutSize(height, filterHeight, stride_height, padding_top, padding_bottom); in groupedConvPrepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 239 int32_t filterHeight, int32_t paddingLeft, in getPaddingScheme() argument
251 calculateExplicitPadding(inHeight, strideHeight, filterHeight, kPaddingSame, in getPaddingScheme()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1846 ANeuralNetworksOperandType filterHeight = scalar; in poolingOpTest() local
1851 strideHeight, filterWidth, filterHeight, activation}, in poolingOpTest()
1858 {input, padImplicit, strideWidth, strideHeight, filterWidth, filterHeight, activation}, in poolingOpTest()
1871 filterHeight, activation, layout}, in poolingOpTest()
1877 filterWidth, filterHeight, activation, layout}, in poolingOpTest()
|