| /frameworks/ml/nn/common/operations/ |
| D | GroupedConv2D.cpp | 49 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvFloat32() argument
69 static_cast<int32_t>(h) * stride_height - padding_top; in groupedConvFloat32()
107 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvQuant8() argument
141 static_cast<int32_t>(h) * stride_height - padding_top; 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
245 static_cast<int32_t>(h) * stride_height - padding_top; 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 …]
|
| D | Conv2D.cpp | 66 int32_t stride_width, stride_height; member
78 stride_height = context->getInputValue<int32_t>(5); in initialize()
94 stride_height = context->getInputValue<int32_t>(8); in initialize()
115 calculateExplicitPadding(input_height, stride_height, dilation_height_factor, in initialize()
124 NN_RET_CHECK_GT(stride_height, 0); in initialize()
180 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
195 convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
205 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
242 stride_width, stride_height, dilation_width_factor, dilation_height_factor, in convNhwc()
254 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
[all …]
|
| D | Pooling.cpp | 46 int32_t stride_width, stride_height; member
60 stride_height = context->getInputValue<int32_t>(6); in initialize()
70 stride_height = context->getInputValue<int32_t>(3); in initialize()
84 calculateExplicitPadding(input_height, stride_height, filter_height, padding_implicit, in initialize()
92 NN_RET_CHECK_GT(stride_height, 0); in initialize()
109 .stride_height = stride_height, in toTfliteParam()
374 uint32_t outHeight = computeOutSize(height, param.filter_height, param.stride_height, in prepare()
|
| /frameworks/ml/nn/runtime/test/specs/V1_0/ |
| D | l2_pool_float_large.mod.py | 22 stride_height = Int32Scalar("stride_height", 1) variable
27 stride_width, stride_height,
|
| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| D | l2_pool_float_large_relaxed.mod.py | 22 stride_height = Int32Scalar("stride_height", 1) variable
27 stride_width, stride_height,
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 148 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
155 int32_t stride_height, int32_t activation, float* outputData,
163 int32_t stride_height, int32_t activation, T* outputData,
172 int32_t stride_width, int32_t stride_height, int32_t numGroups,
|
| D | OperationsUtils.h | 300 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
337 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1528 int32_t stride_width, stride_height; in executeOperation() local
1539 stride_height = getScalarData<int32_t>(operands[ins[8]]); in executeOperation()
1546 stride_height = getScalarData<int32_t>(operands[ins[5]]); in executeOperation()
1574 calculateExplicitPadding(input_height, stride_height, filter_height, in executeOperation()
1580 stride_height, numGroups, &outShape) || in executeOperation()
1592 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1600 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1611 stride_height, numGroups, activation, in executeOperation()
1619 stride_height, numGroups, activation, in executeOperation()
1630 stride_height, numGroups, activation, in executeOperation()
[all …]
|
| D | OperationsUtils.cpp | 715 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvPrepare() argument
753 computeOutSize(height, filterHeight, stride_height, padding_top, padding_bottom); in groupedConvPrepare()
|