| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | Conv2D.cpp | 66 int32_t stride_width, stride_height; member
77 stride_width = context->getInputValue<int32_t>(4); in initialize()
93 stride_width = context->getInputValue<int32_t>(7); in initialize()
123 NN_RET_CHECK_GT(stride_width, 0); in initialize()
197 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
216 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
226 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
280 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
309 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
325 padding_bottom, stride_width, stride_height, dilation_width_factor, in convNhwc()
[all …]
|
| H A D | GroupedConv2D.cpp | 49 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvFloat32() argument
67 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvFloat32()
107 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvQuant8() argument
139 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvQuant8()
186 int32_t stride_width, int32_t stride_height,
195 int32_t stride_width, int32_t stride_height,
205 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvQuant8PerChannel() argument
243 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvQuant8PerChannel()
289 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvFloat16() argument
304 padding_bottom, stride_width, stride_height, numGroups, activation, in groupedConvFloat16()
[all …]
|
| H A D | DepthwiseConv2D.cpp | 52 int32_t stride_width, stride_height; member
64 stride_width = context->getInputValue<int32_t>(4); in initialize()
81 stride_width = context->getInputValue<int32_t>(7); in initialize()
112 NN_RET_CHECK_GT(stride_width, 0); in initialize()
150 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvNhwc()
221 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvNhwc()
592 param.stride_width, param.stride_height, in execute()
605 param.stride_width, param.stride_height, in execute()
624 param.stride_width, param.stride_height, param.dilation_width_factor, in execute()
636 param.stride_width, param.stride_height, in execute()
[all …]
|
| H A D | Pooling.cpp | 48 int32_t stride_width, stride_height; member
61 stride_width = context->getInputValue<int32_t>(5); in initialize()
71 stride_width = context->getInputValue<int32_t>(2); in initialize()
84 calculateExplicitPadding(input_width, stride_width, filter_width, padding_implicit, in initialize()
93 NN_RET_CHECK_GT(stride_width, 0); in initialize()
112 .stride_width = stride_width, in toTfliteParam()
377 uint32_t outWidth = computeOutSize(width, param.filter_width, param.stride_width, in prepare()
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| H A D | l2_pool_float_large.mod.py | 21 stride_width = Int32Scalar("stride_width", 1) variable
27 stride_width, stride_height,
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| H A D | l2_pool_float_large_relaxed.mod.py | 21 stride_width = Int32Scalar("stride_width", 1) variable
27 stride_width, stride_height,
|
| /aosp12/packages/modules/NeuralNetworks/driver/sample/ |
| H A D | SampleDriverFloatXNNPACK.cpp | 345 if (stride_width <= 0) { in CheckConvolutionParams()
364 if (stride_width <= 0) { in CheckDepthwiseConvolutionParams()
390 if (stride_width <= 0) { in CheckPoolingParams()
647 stride_width = getScalarData<int32_t>(operands[ins[5]]); in VisitAveragePool2DNode()
656 stride_width = getScalarData<int32_t>(operands[ins[2]]); in VisitAveragePool2DNode()
722 int32_t stride_width, stride_height, activation; in VisitConv2DNode() local
736 stride_width = getScalarData<int32_t>(operands[ins[7]]); in VisitConv2DNode()
747 stride_width = getScalarData<int32_t>(operands[ins[4]]); in VisitConv2DNode()
774 static_cast<uint32_t>(stride_width), in VisitConv2DNode()
836 stride_width = getScalarData<int32_t>(operands[ins[7]]); in VisitDepthwiseConv2DNode()
[all …]
|
| /aosp12/packages/modules/NeuralNetworks/common/include/ |
| H A D | Operations.h | 152 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
158 int32_t padding_top, int32_t padding_bottom, int32_t stride_width,
166 int32_t padding_top, int32_t padding_bottom, int32_t stride_width,
176 int32_t stride_width, int32_t stride_height, int32_t numGroups,
|
| H A D | OperationsUtils.h | 284 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
321 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
|
| /aosp12/packages/modules/NeuralNetworks/common/ |
| H A D | CpuExecutor.cpp | 1471 int32_t stride_width, stride_height; in executeOperation() local
1481 stride_width = getScalarData<int32_t>(operands[ins[7]]); in executeOperation()
1488 stride_width = getScalarData<int32_t>(operands[ins[4]]); in executeOperation()
1515 calculateExplicitPadding(input_width, stride_width, filter_width, padding_implicit, in executeOperation()
1522 padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
1535 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1543 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1554 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
1562 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
1574 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
[all …]
|
| H A D | OperationsUtils.cpp | 709 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvPrepare() argument
745 computeOutSize(width, filterWidth, stride_width, padding_left, padding_right); in groupedConvPrepare()
|