| /frameworks/ml/nn/common/operations/ |
| D | ResizeImageOps.cpp | 89 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp() argument
91 InputWithLayout<T> input(useNchw); in resizeImageOp()
92 OutputWithLayout<T> output(useNchw); in resizeImageOp()
146 bool useNchw = false; in prepare() local
148 useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
153 uint32_t inHeight = getSizeOfDimension(input, useNchw ? 2 : 1); in prepare()
154 uint32_t inWidth = getSizeOfDimension(input, useNchw ? 3 : 2); in prepare()
155 uint32_t channels = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare()
184 if (useNchw) { in prepare()
195 bool useNchw = false; in execute() local
[all …]
|
| D | Pooling.cpp | 41 bool useNchw = false; member
57 useNchw = context->getInputValue<bool>(10); in initialize()
67 useNchw = context->getInputValue<bool>(7); in initialize()
72 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
73 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
212 InputWithLayout<T> input(param.useNchw); in averagePool()
213 OutputWithLayout<T> output(param.useNchw); in averagePool()
225 InputWithLayout<T> input(param.useNchw); in l2Pool()
226 OutputWithLayout<T> output(param.useNchw); in l2Pool()
238 InputWithLayout<T> input(param.useNchw); in maxPool()
[all …]
|
| D | RoiPooling.cpp | 140 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling() argument
141 InputWithLayout<T_Input> input(useNchw); in roiPooling()
142 OutputWithLayout<T_Input> output(useNchw); in roiPooling()
157 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling() argument
162 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
203 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
211 uint32_t inHeight = getSizeOfDimension(input, useNchw ? 2 : 1); in prepare()
212 uint32_t inWidth = getSizeOfDimension(input, useNchw ? 3 : 2); in prepare()
213 uint32_t inDepth = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare()
239 if (useNchw) { in prepare()
|
| D | Conv2D.cpp | 56 bool useNchw = false; member
68 useNchw = context->getInputValue<bool>(7); in initialize()
84 useNchw = context->getInputValue<bool>(10); in initialize()
96 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
97 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
267 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv() argument
269 InputWithLayout<T_Input> input(useNchw); in conv()
270 OutputWithLayout<T_Input> output(useNchw); in conv()
381 int32_t dilationHeightFactor, int32_t activation, bool useNchw, in convQuant8PerChannel() argument
383 InputWithLayout<uint8_t> input(useNchw); in convQuant8PerChannel()
[all …]
|
| D | TransposeConv2D.cpp | 54 bool useNchw = false; member
64 useNchw = context->getInputValue<bool>(8); in initialize()
71 int32_t outputWidth = useNchw ? outputShapeData[3] : outputShapeData[2]; in initialize()
72 int32_t outputHeight = useNchw ? outputShapeData[2] : outputShapeData[1]; in initialize()
85 useNchw = context->getInputValue<bool>(10); in initialize()
294 InputWithLayout<T_Input> input(param.useNchw); in transposeConv()
295 OutputWithLayout<T_Input> output(param.useNchw); in transposeConv()
412 InputWithLayout<uint8_t> input(param.useNchw); in transposeConvQuant8PerChannel()
413 OutputWithLayout<uint8_t> output(param.useNchw); in transposeConvQuant8PerChannel()
488 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
[all …]
|
| D | RoiAlign.cpp | 314 int32_t heightSamplingRatio, int32_t widthSamplingRatio, bool useNchw, in roiAlign() argument
316 InputWithLayout<T_Input> input(useNchw); in roiAlign()
317 OutputWithLayout<T_Input> output(useNchw); in roiAlign()
368 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
376 uint32_t inHeight = getSizeOfDimension(input, useNchw ? 2 : 1); in prepare()
377 uint32_t inWidth = getSizeOfDimension(input, useNchw ? 3 : 2); in prepare()
378 uint32_t inDepth = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare()
415 if (useNchw) { in prepare()
|
| D | InstanceNormalization.cpp | 79 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm() argument
80 InputWithLayout<T> input(useNchw); in instanceNorm()
81 OutputWithLayout<T> output(useNchw); in instanceNorm()
|
| D | GenerateProposals.cpp | 951 bool useNchw, std::vector<float>* scoresOutData, in generateProposalsFloat32Compute() argument
954 InputWithLayout<float> score_nhwc(useNchw), delta_nhwc(useNchw); in generateProposalsFloat32Compute()
969 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat32() argument
976 iouThreshold, minSize, useNchw, &scoresOut_float32, &roiOut_float32, &batchesOut)); in generateProposalsFloat32()
1012 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat16() argument
1027 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsFloat16()
1060 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsQuant() argument
1077 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsQuant()
1165 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
1181 uint32_t height = getSizeOfDimension(scoreShape, useNchw ? 2 : 1); in prepare()
[all …]
|
| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | BoundingBox.cpp | 40 bool useNchw; in roiConstructor() local
42 useNchw = op->inputs[9]->value<bool8>(); in roiConstructor()
44 useNchw = op->inputs[7]->value<bool8>(); in roiConstructor()
52 auto outDepth = op->inputs[0]->dimensions[useNchw ? 1 : 3]; in roiConstructor()
55 if (useNchw) { in roiConstructor()
79 bool useNchw; in roiFinalizer() local
81 useNchw = op->inputs[9]->value<bool8>(); in roiFinalizer()
83 useNchw = op->inputs[7]->value<bool8>(); in roiFinalizer()
87 uint32_t height = op->inputs[0]->dimensions[useNchw ? 2 : 1].getValue(); in roiFinalizer()
88 uint32_t width = op->inputs[0]->dimensions[useNchw ? 3 : 2].getValue(); in roiFinalizer()
[all …]
|
| D | Convolutions.cpp | 35 bool useNchw = false; in conv2DExplicitConstructor() local
38 useNchw = op->inputs[10]->value<bool8>(); in conv2DExplicitConstructor()
44 int heightIndex = useNchw ? 2 : 1; in conv2DExplicitConstructor()
45 int widthIndex = useNchw ? 3 : 2; in conv2DExplicitConstructor()
46 int channelIndex = useNchw ? 1 : 3; in conv2DExplicitConstructor()
86 bool useNchw = false; in conv2DImplicitConstructor() local
89 useNchw = op->inputs[7]->value<bool8>(); in conv2DImplicitConstructor()
95 int heightIndex = useNchw ? 2 : 1; in conv2DImplicitConstructor()
96 int widthIndex = useNchw ? 3 : 2; in conv2DImplicitConstructor()
97 int channelIndex = useNchw ? 1 : 3; in conv2DImplicitConstructor()
[all …]
|
| D | Poolings.cpp | 36 bool useNchw = false; in poolingExplicitOpConstructor() local
37 if (op->inputs.size() > 10) useNchw = op->inputs[10]->value<bool8>(); in poolingExplicitOpConstructor()
38 int heightIndex = useNchw ? 2 : 1; in poolingExplicitOpConstructor()
39 int widthIndex = useNchw ? 3 : 2; in poolingExplicitOpConstructor()
40 int channelIndex = useNchw ? 1 : 3; in poolingExplicitOpConstructor()
75 bool useNchw = false; in poolingImplicitOpConstructor() local
76 if (op->inputs.size() > 7) useNchw = op->inputs[7]->value<bool8>(); in poolingImplicitOpConstructor()
77 int heightIndex = useNchw ? 2 : 1; in poolingImplicitOpConstructor()
78 int widthIndex = useNchw ? 3 : 2; in poolingImplicitOpConstructor()
79 int channelIndex = useNchw ? 1 : 3; in poolingImplicitOpConstructor()
|
| D | Reshape.cpp | 26 bool useNchw = false; in spaceToDepthConstructor() local
27 if (op->inputs.size() > 2) useNchw = op->inputs[2]->value<bool8>(); in spaceToDepthConstructor()
28 int heightIndex = useNchw ? 2 : 1; in spaceToDepthConstructor()
29 int widthIndex = useNchw ? 3 : 2; in spaceToDepthConstructor()
30 int depthIndex = useNchw ? 1 : 3; in spaceToDepthConstructor()
39 if (useNchw) { in spaceToDepthConstructor()
79 bool useNchw = false; in depthToSpaceConstructor() local
80 if (op->inputs.size() > 2) useNchw = op->inputs[2]->value<bool8>(); in depthToSpaceConstructor()
81 int heightIndex = useNchw ? 2 : 1; in depthToSpaceConstructor()
82 int widthIndex = useNchw ? 3 : 2; in depthToSpaceConstructor()
[all …]
|
| D | Resize.cpp | 27 bool useNchw = op->inputs.size() > 3 ? static_cast<bool>(op->inputs[3]->value<bool8>()) : false; in resizeOpConstructor() local
28 int heightIndex = useNchw ? 2 : 1; in resizeOpConstructor()
29 int widthIndex = useNchw ? 3 : 2; in resizeOpConstructor()
54 if (useNchw) { in resizeOpConstructor()
|
| /frameworks/ml/nn/common/include/ |
| D | CpuOperationUtils.h | 141 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout() argument
165 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout() argument
|