| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Conv2D.cpp | 64 int32_t stride_width, stride_height; member
75 stride_width = context->getInputValue<int32_t>(4); in initialize()
91 stride_width = context->getInputValue<int32_t>(7); in initialize()
111 calculateExplicitPadding(input_width, stride_width, dilation_width_factor, filter_width, in initialize()
121 NN_RET_CHECK_GT(stride_width, 0); in initialize()
175 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData() argument
187 stride_width != 1 || stride_height != 1 || filter_width != 1 || filter_height != 1; in needim2colData()
194 int32_t /*padding_bottom*/, int32_t stride_width, int32_t stride_height, in convNhwc() argument
208 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc()
213 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
[all …]
|
| D | GroupedConv2D.cpp | 53 int32_t /*padding_bottom*/, int32_t stride_width, int32_t stride_height, in groupedConvFloat32() argument
71 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvFloat32()
111 int32_t /*padding_bottom*/, int32_t stride_width, int32_t stride_height, in groupedConvQuant8() argument
143 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvQuant8()
190 int32_t stride_width, int32_t stride_height,
199 int32_t stride_width, int32_t stride_height,
209 int32_t /*padding_bottom*/, int32_t stride_width, in groupedConvQuant8PerChannel() argument
248 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvQuant8PerChannel()
294 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvFloat16() argument
309 padding_bottom, stride_width, stride_height, numGroups, activation, in groupedConvFloat16()
[all …]
|
| D | DepthwiseConv2D.cpp | 48 int32_t stride_width, stride_height; member
60 stride_width = context->getInputValue<int32_t>(4); in initialize()
77 stride_width = context->getInputValue<int32_t>(7); in initialize()
98 calculateExplicitPadding(input_width, stride_width, dilation_width_factor, filter_width, in initialize()
108 NN_RET_CHECK_GT(stride_width, 0); in initialize()
146 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvNhwc()
217 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvNhwc()
462 computeOutSize(width, filterWidth, param.stride_width, param.dilation_width_factor, in prepare()
489 param.stride_width, param.stride_height, in execute()
502 param.stride_width, param.stride_height, in execute()
[all …]
|
| D | Pooling.cpp | 50 int32_t stride_width, stride_height; member
63 stride_width = context->getInputValue<int32_t>(5); in initialize()
73 stride_width = context->getInputValue<int32_t>(2); in initialize()
86 calculateExplicitPadding(input_width, stride_width, filter_width, padding_implicit, in initialize()
95 NN_RET_CHECK_GT(stride_width, 0); in initialize()
114 .stride_width = stride_width, in toTfliteParam()
313 uint32_t outWidth = computeOutSize(width, param.filter_width, param.stride_width, in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | l2_pool_float_large.mod.py | 21 stride_width = Int32Scalar("stride_width", 1) variable
27 stride_width, stride_height,
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| D | l2_pool_float_large_relaxed.mod.py | 21 stride_width = Int32Scalar("stride_width", 1) variable
27 stride_width, stride_height,
|
| /packages/modules/NeuralNetworks/driver/sample_hidl/ |
| D | SampleDriverFloatXNNPACK.cpp | 342 static V1_3::ErrorStatus CheckConvolutionParams(int32_t stride_width, int32_t stride_height, in CheckConvolutionParams() argument
345 if (stride_width <= 0) { in CheckConvolutionParams()
362 int32_t stride_width, int32_t stride_height, int32_t dilation_width_factor, in CheckDepthwiseConvolutionParams() argument
364 if (stride_width <= 0) { in CheckDepthwiseConvolutionParams()
388 static V1_3::ErrorStatus CheckPoolingParams(int32_t stride_width, int32_t stride_height, in CheckPoolingParams() argument
390 if (stride_width <= 0) { in CheckPoolingParams()
403 if (filter_width == 1 && filter_height == 1 && std::max(stride_width, stride_height) > 1) { in CheckPoolingParams()
635 int32_t stride_width, stride_height, filter_width, filter_height, activation; in VisitAveragePool2DNode() local
647 stride_width = getScalarData<int32_t>(operands[ins[5]]); in VisitAveragePool2DNode()
656 stride_width = getScalarData<int32_t>(operands[ins[2]]); in VisitAveragePool2DNode()
[all …]
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 135 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
141 int32_t padding_top, int32_t padding_bottom, int32_t stride_width,
149 int32_t padding_top, int32_t padding_bottom, int32_t stride_width,
159 int32_t stride_width, int32_t stride_height, int32_t numGroups,
|
| D | OperationsExecutionUtils.h | 226 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
263 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1472 int32_t stride_width, stride_height; in executeOperation() local
1482 stride_width = getScalarData<int32_t>(operands[ins[7]]); in executeOperation()
1489 stride_width = getScalarData<int32_t>(operands[ins[4]]); in executeOperation()
1516 calculateExplicitPadding(input_width, stride_width, filter_width, padding_implicit, in executeOperation()
1523 padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
1536 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1544 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1555 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
1563 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
1575 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
[all …]
|
| D | OperationsExecutionUtils.cpp | 618 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvPrepare() argument
654 computeOutSize(width, filterWidth, stride_width, padding_left, padding_right); in groupedConvPrepare()
|