| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | l2_pool_float_large_relaxed.model.cpp | 10 auto stride_width = model->addOperand(&type1); in CreateModel() local
20 model->setOperandValue(stride_width, stride_width_init, sizeof(int32_t) * 1); in CreateModel()
29 …model->addOperation(ANEURALNETWORKS_L2_POOL_2D, {op1, pad0, pad0, pad0, pad0, stride_width, stride… in CreateModel()
51 auto stride_width = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
61 model->setOperandValue(stride_width, stride_width_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …model->addOperation(ANEURALNETWORKS_L2_POOL_2D, {op1, pad0, pad0, pad0, pad0, stride_width, stride… in CreateModel_dynamic_output_shape()
|
| D | l2_pool_float_large.model.cpp | 10 auto stride_width = model->addOperand(&type1); in CreateModel() local
20 model->setOperandValue(stride_width, stride_width_init, sizeof(int32_t) * 1); in CreateModel()
29 …model->addOperation(ANEURALNETWORKS_L2_POOL_2D, {op1, pad0, pad0, pad0, pad0, stride_width, stride… in CreateModel()
49 auto stride_width = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
59 model->setOperandValue(stride_width, stride_width_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
68 …model->addOperation(ANEURALNETWORKS_L2_POOL_2D, {op1, pad0, pad0, pad0, pad0, stride_width, stride… in CreateModel_dynamic_output_shape()
|
| /frameworks/ml/nn/common/operations/ |
| D | Conv2D.cpp | 53 int32_t stride_width, stride_height; member
64 stride_width = context->getInputValue<int32_t>(4); in initialize()
80 stride_width = context->getInputValue<int32_t>(7); in initialize()
100 calculateExplicitPadding(input_width, stride_width, dilation_width_factor, filter_width, in initialize()
110 NN_RET_CHECK_GT(stride_width, 0); in initialize()
167 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
182 convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
192 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
229 stride_width, stride_height, dilation_width_factor, dilation_height_factor, in convNhwc()
239 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
[all …]
|
| D | GroupedConv2D.cpp | 45 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvFloat32() argument
63 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvFloat32()
102 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvQuant8() argument
134 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvQuant8()
181 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvQuant8PerChannel() argument
220 static_cast<int32_t>(w) * stride_width - padding_left; in groupedConvQuant8PerChannel()
266 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvFloat16() argument
281 padding_bottom, stride_width, stride_height, numGroups, activation, in groupedConvFloat16()
|
| D | Pooling.cpp | 38 int32_t stride_width, stride_height; member
51 stride_width = context->getInputValue<int32_t>(5); in initialize()
61 stride_width = context->getInputValue<int32_t>(2); in initialize()
74 calculateExplicitPadding(input_width, stride_width, filter_width, padding_implicit, in initialize()
83 NN_RET_CHECK_GT(stride_width, 0); in initialize()
98 .stride_width = stride_width, in toTfliteParam()
317 uint32_t outWidth = computeOutSize(width, param.filter_width, param.stride_width, in prepare()
|
| D | DepthwiseConv2D.cpp | 81 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvFloat32()
125 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvQuant8()
|
| /frameworks/ml/nn/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,
|
| /frameworks/ml/nn/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,
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 145 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
151 int32_t padding_top, int32_t padding_bottom, int32_t stride_width,
158 int32_t padding_top, int32_t padding_bottom, int32_t stride_width,
167 int32_t stride_width, int32_t stride_height, int32_t numGroups,
|
| D | OperationsUtils.h | 298 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
370 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 754 int32_t stride_width, stride_height; in executeOperation() local
763 stride_width = getScalarData<int32_t>(mOperands[ins[4]]); in executeOperation()
780 stride_width = getScalarData<int32_t>(mOperands[ins[7]]); in executeOperation()
815 calculateExplicitPadding(input_width, stride_width, dilation_width_factor, in executeOperation()
824 padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
837 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
845 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
855 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
864 padding_left, padding_right, padding_top, padding_bottom, stride_width, in executeOperation()
1556 int32_t stride_width, stride_height; in executeOperation() local
[all …]
|
| D | OperationsUtils.cpp | 352 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in depthwiseConvPrepare() argument
387 uint32_t outWidth = computeOutSize(width, filterWidth, stride_width, dilation_width_factor, in depthwiseConvPrepare()
868 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvPrepare() argument
902 computeOutSize(width, filterWidth, stride_width, padding_left, padding_right); in groupedConvPrepare()
|