| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Convolutions.cpp | 35 int32_t strideHeight = op->inputs[8]->value<int32_t>(); in conv2DExplicitConstructor() local
69 strideHeight, dilationHeight, paddingTop, paddingBottom, in conv2DExplicitConstructor()
87 int32_t strideHeight = op->inputs[5]->value<int32_t>(); in conv2DImplicitConstructor() local
121 strideHeight, dilationHeight, paddingScheme, in conv2DImplicitConstructor()
274 int32_t strideHeight = op->inputs[8]->value<int32_t>(); in depthwiseConv2DExplicitConstructor() local
309 strideHeight, dilationHeight, paddingTop, paddingBottom, in depthwiseConv2DExplicitConstructor()
327 int32_t strideHeight = op->inputs[5]->value<int32_t>(); in depthwiseConv2DImplicitConstructor() local
362 strideHeight, dilationHeight, paddingScheme, in depthwiseConv2DImplicitConstructor()
527 int32_t strideHeight = op->inputs[8]->value<int32_t>(); in groupedConv2DExplicitConstructor() local
562 strideHeight, /*dilation=*/1, paddingTop, paddingBottom, in groupedConv2DExplicitConstructor()
[all …]
|
| D | Poolings.cpp | 33 int32_t strideHeight = op->inputs[6]->value<int32_t>(); in poolingExplicitOpConstructor() local
54 explicitPadding(op->inputs[0]->dimensions[heightIndex], filterHeight, strideHeight, in poolingExplicitOpConstructor()
72 int32_t strideHeight = op->inputs[3]->value<int32_t>(); in poolingImplicitOpConstructor() local
93 implicitPadding(op->inputs[0]->dimensions[heightIndex], filterHeight, strideHeight, in poolingImplicitOpConstructor()
|
| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 134 int32_t paddingBottom, int32_t strideWidth, int32_t strideHeight, in depthwiseConvNhwc() argument
149 .stride_height = static_cast<int16>(strideHeight), in depthwiseConvNhwc()
169 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor, in depthwiseConvNhwc() argument
183 paddingBottom, strideWidth, strideHeight, dilationWidthFactor, in depthwiseConvNhwc()
194 int32_t paddingBottom, int32_t strideWidth, int32_t strideHeight, in depthwiseConvNhwc() argument
220 .stride_height = static_cast<int16>(strideHeight), in depthwiseConvNhwc()
245 int32_t paddingBottom, int32_t strideWidth, int32_t strideHeight, in depthwiseConvNhwc() argument
264 paddingTop, paddingBottom, strideWidth, strideHeight, in depthwiseConvNhwc()
278 int32_t paddingBottom, int32_t strideWidth, int32_t strideHeight, in depthwiseConvQuant8PerChannelNhwc() argument
329 int32_t hInputOrigin = static_cast<int32_t>(h) * strideHeight - paddingTop; in depthwiseConvQuant8PerChannelNhwc()
[all …]
|
| D | TransposeConv2D.cpp | 62 int32_t strideWidth, strideHeight; member
72 strideHeight = context->getInputValue<int32_t>(6); in initialize()
85 calculateExplicitPaddingTransposeConv(outputHeight, strideHeight, filterHeight, in initialize()
93 strideHeight = context->getInputValue<int32_t>(8); in initialize()
104 NN_RET_CHECK_GT(strideHeight, 0); in initialize()
122 int32_t strideWidth = param.strideWidth, strideHeight = param.strideHeight; \
143 int32_t hOutputOrigin = static_cast<int32_t>(h) * strideHeight - paddingTop; in transposeConvNhwc()
230 int32_t hOutputOrigin = static_cast<int32_t>(h) * strideHeight - paddingTop; in transposeConvNhwc()
372 int32_t hOutputOrigin = static_cast<int32_t>(h) * strideHeight - paddingTop; in transposeConvQuant8PerChannelNhwc()
526 uint32_t outHeight = computeOutSizeTransposeConv(height, filterHeight, param.strideHeight, in prepare()
|
| D | Conv2D.cpp | 331 int32_t strideHeight, int32_t dilationWidthFactor, in convQuant8PerChannelNhwc() argument
378 int32_t hInputOrigin = static_cast<int32_t>(h) * strideHeight - paddingTop; in convQuant8PerChannelNhwc()
424 int32_t strideHeight, int32_t dilationWidthFactor, in convQuant8PerChannelNhwc() argument
464 convParams.stride_height = strideHeight; in convQuant8PerChannelNhwc()
487 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor, in convQuant8PerChannel() argument
497 strideHeight, dilationWidthFactor, dilationHeightFactor, activation, in convQuant8PerChannel()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 52 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor,
58 int32_t paddingBottom, int32_t strideWidth, int32_t strideHeight,
66 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor,
74 int32_t strideWidth, int32_t strideHeight,
|
| D | OperationsUtils.h | 238 int32_t strideHeight, int32_t filterWidth, in getPaddingScheme() argument
251 calculateExplicitPadding(inHeight, strideHeight, filterHeight, kPaddingSame, in getPaddingScheme()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1843 ANeuralNetworksOperandType strideHeight = scalar; in poolingOpTest() local
1850 strideHeight, filterWidth, filterHeight, activation}, in poolingOpTest()
1857 {input, padImplicit, strideWidth, strideHeight, filterWidth, filterHeight, activation}, in poolingOpTest()
1869 {input, padLeft, padRight, padTop, padBottom, strideWidth, strideHeight, filterWidth, in poolingOpTest()
1875 {input, padImplicit, strideWidth, strideHeight, in poolingOpTest()
2134 ANeuralNetworksOperandType strideHeight = scalar; in convOpTest() local
2141 strideWidth, strideHeight, activation}, in convOpTest()
2151 {input, filter, bias, padImplicit, strideWidth, strideHeight, activation}, {output}, in convOpTest()
2166 {input, filter, bias, padLeft, padRight, padTop, padBottom, strideWidth, strideHeight, in convOpTest()
2176 {input, filter, bias, padImplicit, strideWidth, strideHeight, activation, layout}, in convOpTest()
[all …]
|