| /frameworks/ml/nn/common/operations/ |
| 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()
177 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData() argument
189 stride_width != 1 || stride_height != 1 || filter_width != 1 || filter_height != 1; in needim2colData()
196 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
210 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc()
215 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
[all …]
|
| 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 | DepthwiseConv2D.cpp | 50 int32_t stride_width, stride_height; member
63 stride_height = context->getInputValue<int32_t>(5); in initialize()
80 stride_height = context->getInputValue<int32_t>(8); in initialize()
102 calculateExplicitPadding(input_height, stride_height, dilation_height_factor, in initialize()
111 NN_RET_CHECK_GT(stride_height, 0); in initialize()
149 .stride_height = static_cast<int16>(strideHeight), in depthwiseConvNhwc()
220 .stride_height = static_cast<int16>(strideHeight), in depthwiseConvNhwc()
554 computeOutSize(height, filterHeight, param.stride_height, param.dilation_height_factor, in prepare()
584 param.stride_width, param.stride_height, in execute()
597 param.stride_width, param.stride_height, in execute()
[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 | 1527 int32_t stride_width, stride_height; in executeOperation() local
1538 stride_height = getScalarData<int32_t>(operands[ins[8]]); in executeOperation()
1545 stride_height = getScalarData<int32_t>(operands[ins[5]]); in executeOperation()
1573 calculateExplicitPadding(input_height, stride_height, filter_height, in executeOperation()
1579 stride_height, numGroups, &outShape) || in executeOperation()
1591 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1599 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1610 stride_height, numGroups, activation, in executeOperation()
1618 stride_height, numGroups, activation, in executeOperation()
1629 stride_height, numGroups, activation, in executeOperation()
[all …]
|
| D | OperationsUtils.cpp | 711 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvPrepare() argument
749 computeOutSize(height, filterHeight, stride_height, padding_top, padding_bottom); in groupedConvPrepare()
|