| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Convolutions.cpp | 34 int32_t strideWidth = op->inputs[7]->value<int32_t>(); in conv2DExplicitConstructor() local
74 strideWidth, dilationWidth, paddingLeft, paddingRight, in conv2DExplicitConstructor()
86 int32_t strideWidth = op->inputs[4]->value<int32_t>(); in conv2DImplicitConstructor() local
124 strideWidth, dilationWidth, paddingScheme, in conv2DImplicitConstructor()
273 int32_t strideWidth = op->inputs[7]->value<int32_t>(); in depthwiseConv2DExplicitConstructor() local
314 strideWidth, dilationWidth, paddingLeft, paddingRight, in depthwiseConv2DExplicitConstructor()
326 int32_t strideWidth = op->inputs[4]->value<int32_t>(); in depthwiseConv2DImplicitConstructor() local
365 strideWidth, dilationWidth, paddingScheme, in depthwiseConv2DImplicitConstructor()
526 int32_t strideWidth = op->inputs[7]->value<int32_t>(); in groupedConv2DExplicitConstructor() local
567 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/operations/ |
| D | TransposeConv2D.cpp | 62 int32_t strideWidth, strideHeight; member
71 strideWidth = context->getInputValue<int32_t>(5); in initialize()
83 calculateExplicitPaddingTransposeConv(outputWidth, strideWidth, filterWidth, in initialize()
92 strideWidth = context->getInputValue<int32_t>(7); in initialize()
103 NN_RET_CHECK_GT(strideWidth, 0); in initialize()
122 int32_t strideWidth = param.strideWidth, strideHeight = param.strideHeight; \
142 int32_t wOutputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in transposeConvNhwc()
229 int32_t wOutputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in transposeConvNhwc()
371 int32_t wOutputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in transposeConvQuant8PerChannelNhwc()
524 uint32_t outWidth = computeOutSizeTransposeConv(width, filterWidth, param.strideWidth, in prepare()
|
| D | Conv2D.cpp | 330 int32_t paddingTop, int32_t paddingBottom, int32_t strideWidth, in convQuant8PerChannelNhwc() argument
377 int32_t wInputOrigin = static_cast<int32_t>(w) * strideWidth - paddingLeft; in convQuant8PerChannelNhwc()
423 int32_t paddingTop, int32_t paddingBottom, int32_t strideWidth, in convQuant8PerChannelNhwc() argument
465 convParams.stride_width = strideWidth; in convQuant8PerChannelNhwc()
487 int32_t strideWidth, int32_t strideHeight, int32_t dilationWidthFactor, in convQuant8PerChannel() argument
496 biasData, biasShape, paddingLeft, paddingRight, paddingTop, paddingBottom, strideWidth, 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 | 237 inline PaddingScheme getPaddingScheme(int32_t inWidth, int32_t inHeight, int32_t strideWidth, in getPaddingScheme() argument
249 calculateExplicitPadding(inWidth, strideWidth, filterWidth, kPaddingSame, &expectedPaddingLeft, in getPaddingScheme()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1842 ANeuralNetworksOperandType strideWidth = scalar; in poolingOpTest() local
1849 {input, padLeft, padRight, padTop, padBottom, strideWidth, 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()
2133 ANeuralNetworksOperandType strideWidth = 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 …]
|