| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 32 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor, in depthwiseConvFloat16() argument
46 paddingBottom, strideWidth, strideHeight, dilationWidthFactor, in depthwiseConvFloat16()
68 int32_t paddingBottom, int32_t strideWidth, int32_t strideHeight, in depthwiseConvFloat32() argument
81 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvFloat32()
102 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor, in depthwiseConvQuant8() argument
125 .stride_width = static_cast<int16>(strideWidth), in depthwiseConvQuant8()
151 int32_t strideWidth, int32_t strideHeight, in depthwiseConvQuant8PerChannel() argument
202 int32_t wInputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in depthwiseConvQuant8PerChannel()
|
| D | TransposeConv2D.cpp | 52 int32_t strideWidth, strideHeight; member
61 strideWidth = context->getInputValue<int32_t>(5); in initialize()
73 calculateExplicitPaddingTransposeConv(outputWidth, strideWidth, filterWidth, in initialize()
82 strideWidth = context->getInputValue<int32_t>(7); in initialize()
93 NN_RET_CHECK_GT(strideWidth, 0); in initialize()
112 int32_t strideWidth = param.strideWidth, strideHeight = param.strideHeight; \
132 int32_t wOutputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in transposeConvNhwc()
219 int32_t wOutputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in transposeConvNhwc()
360 int32_t wOutputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in transposeConvQuant8PerChannelNhwc()
504 uint32_t outWidth = computeOutSizeTransposeConv(width, filterWidth, param.strideWidth, in prepare()
|
| D | Conv2D.cpp | 286 int32_t paddingTop, int32_t paddingBottom, int32_t strideWidth, in convQuant8PerChannelNhwc() argument
333 int32_t wInputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in convQuant8PerChannelNhwc()
379 int32_t paddingTop, int32_t paddingBottom, int32_t strideWidth, in convQuant8PerChannel() argument
389 biasData, biasShape, paddingLeft, paddingRight, paddingTop, paddingBottom, strideWidth, in convQuant8PerChannel()
|
| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Convolutions.cpp | 33 int32_t strideWidth = op->inputs[7]->value<int32_t>(); in conv2DExplicitConstructor() local
73 strideWidth, dilationWidth, paddingLeft, paddingRight, in conv2DExplicitConstructor()
84 int32_t strideWidth = op->inputs[4]->value<int32_t>(); in conv2DImplicitConstructor() local
122 strideWidth, dilationWidth, paddingScheme, in conv2DImplicitConstructor()
269 int32_t strideWidth = op->inputs[7]->value<int32_t>(); in depthwiseConv2DExplicitConstructor() local
310 strideWidth, dilationWidth, paddingLeft, paddingRight, in depthwiseConv2DExplicitConstructor()
322 int32_t strideWidth = op->inputs[4]->value<int32_t>(); in depthwiseConv2DImplicitConstructor() local
362 strideWidth, dilationWidth, paddingScheme, in depthwiseConv2DImplicitConstructor()
516 int32_t strideWidth = op->inputs[7]->value<int32_t>(); in groupedConv2DExplicitConstructor() local
557 strideWidth, /*dilation=*/1, paddingLeft, paddingRight, in groupedConv2DExplicitConstructor()
[all …]
|
| D | Poolings.cpp | 32 int32_t strideWidth = op->inputs[5]->value<int32_t>(); in poolingExplicitOpConstructor() local
59 explicitPadding(op->inputs[0]->dimensions[widthIndex], filterWidth, strideWidth, /*dilation=*/1, in poolingExplicitOpConstructor()
71 int32_t strideWidth = op->inputs[2]->value<int32_t>(); in poolingImplicitOpConstructor() local
95 implicitPadding(op->inputs[0]->dimensions[widthIndex], filterWidth, strideWidth, in poolingImplicitOpConstructor()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 53 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor,
59 int32_t paddingBottom, int32_t strideWidth, int32_t strideHeight,
67 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor,
75 int32_t strideWidth, int32_t strideHeight,
|
| D | OperationsUtils.h | 234 int32_t strideWidth, int32_t strideHeight, in getPaddingScheme() argument
245 calculateExplicitPadding(inWidth, strideWidth, filterWidth, kPaddingSame, in getPaddingScheme()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1008 ANeuralNetworksOperandType strideWidth = scalar; in poolingOpTest() local
1015 {input, padLeft, padRight, padTop, padBottom, strideWidth, in poolingOpTest()
1023 {input, padImplicit, strideWidth, strideHeight, filterWidth, filterHeight, activation}, in poolingOpTest()
1035 {input, padLeft, padRight, padTop, padBottom, strideWidth, strideHeight, filterWidth, in poolingOpTest()
1041 {input, padImplicit, strideWidth, strideHeight, in poolingOpTest()
1261 ANeuralNetworksOperandType strideWidth = scalar; in convOpTest() local
1269 strideWidth, strideHeight, activation}, in convOpTest()
1279 {input, filter, bias, padImplicit, strideWidth, strideHeight, activation}, {output}); in convOpTest()
1293 {input, filter, bias, padLeft, padRight, padTop, padBottom, strideWidth, strideHeight, in convOpTest()
1303 {input, filter, bias, padImplicit, strideWidth, strideHeight, activation, layout}, in convOpTest()
[all …]
|