Home
last modified time | relevance | path

Searched refs:widthIndex (Results 1 – 4 of 4) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
H A DConvolutions.cpp46 int widthIndex = useNchw ? 3 : 2; in conv2DExplicitConstructor() local
75 &op->outputs[0]->dimensions[widthIndex]); in conv2DExplicitConstructor()
98 int widthIndex = useNchw ? 3 : 2; in conv2DImplicitConstructor() local
125 &op->outputs[0]->dimensions[widthIndex]); in conv2DImplicitConstructor()
285 int widthIndex = useNchw ? 3 : 2; in depthwiseConv2DExplicitConstructor() local
315 &op->outputs[0]->dimensions[widthIndex]); in depthwiseConv2DExplicitConstructor()
338 int widthIndex = useNchw ? 3 : 2; in depthwiseConv2DImplicitConstructor() local
530 int widthIndex = useNchw ? 3 : 2; in groupedConv2DExplicitConstructor() local
582 int widthIndex = useNchw ? 3 : 2; in groupedConv2DImplicitConstructor() local
684 int widthIndex = useNchw ? 3 : 2; in transposeConv2DExplicitConstructor() local
[all …]
H A DPoolings.cpp39 int widthIndex = useNchw ? 3 : 2; in poolingExplicitOpConstructor() local
59 explicitPadding(op->inputs[0]->dimensions[widthIndex], filterWidth, strideWidth, /*dilation=*/1, in poolingExplicitOpConstructor()
60 paddingLeft, paddingRight, &op->outputs[0]->dimensions[widthIndex]); in poolingExplicitOpConstructor()
78 int widthIndex = useNchw ? 3 : 2; in poolingImplicitOpConstructor() local
95 implicitPadding(op->inputs[0]->dimensions[widthIndex], filterWidth, strideWidth, in poolingImplicitOpConstructor()
96 /*dilation=*/1, paddingScheme, &op->outputs[0]->dimensions[widthIndex]); in poolingImplicitOpConstructor()
H A DResize.cpp29 int widthIndex = useNchw ? 3 : 2; in resizeOpConstructor() local
41 outWidth = op->inputs[0]->dimensions[widthIndex] * op->inputs[1]->value<float>(); in resizeOpConstructor()
45 outWidth = op->inputs[0]->dimensions[widthIndex] * in resizeOpConstructor()
H A DReshape.cpp32 int widthIndex = useNchw ? 3 : 2; in spaceToDepthConstructor() local
39 auto outWidth = op->inputs[0]->dimensions[widthIndex].exactDiv(blockSize); in spaceToDepthConstructor()
87 int widthIndex = useNchw ? 3 : 2; in depthToSpaceConstructor() local
94 auto outWidth = op->inputs[0]->dimensions[widthIndex] * blockSize; in depthToSpaceConstructor()
175 int widthIndex = useNchw ? 3 : 2; in batchToSpaceConstructor() local
183 auto outWidth = op->inputs[0]->dimensions[widthIndex] * blockWidth; in batchToSpaceConstructor()
233 int widthIndex = useNchw ? 3 : 2; in spaceToBatchConstructor() local
241 auto paddedWidth = op->inputs[0]->dimensions[widthIndex] + in spaceToBatchConstructor()