| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Convolutions.cpp | 35 bool useNchw = false; in conv2DExplicitConstructor() local
86 bool useNchw = false; in conv2DImplicitConstructor() local
271 bool useNchw = false; in depthwiseConv2DExplicitConstructor() local
324 bool useNchw = false; in depthwiseConv2DImplicitConstructor() local
518 bool useNchw = op->inputs[11]->value<bool8>(); in groupedConv2DExplicitConstructor() local
570 bool useNchw = op->inputs[8]->value<bool8>(); in groupedConv2DImplicitConstructor() local
668 bool useNchw = op->inputs[10]->value<bool8>(); in transposeConv2DExplicitConstructor() local
711 bool useNchw = op->inputs[8]->value<bool8>(); in transposeConv2DImplicitConstructor() local
|
| D | BoundingBox.cpp | 40 bool useNchw; in roiConstructor() local
79 bool useNchw; in roiFinalizer() local
164 bool useNchw = op->inputs[2]->value<bool8>(); in heatmapMaxKeypointConstructor() local
|
| D | Poolings.cpp | 36 bool useNchw = false; in poolingExplicitOpConstructor() local
75 bool useNchw = false; in poolingImplicitOpConstructor() local
|
| D | Resize.cpp | 27 bool useNchw = op->inputs.size() > 3 ? static_cast<bool>(op->inputs[3]->value<bool8>()) : false; in resizeOpConstructor() local
|
| D | Reshape.cpp | 26 bool useNchw = false; in spaceToDepthConstructor() local
79 bool useNchw = false; in depthToSpaceConstructor() local
166 bool useNchw = false; in batchToSpaceConstructor() local
222 bool useNchw = false; in spaceToBatchConstructor() local
|
| /frameworks/ml/nn/common/operations/ |
| D | ResizeImageOps.cpp | 89 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp()
146 bool useNchw = false; in prepare() local
195 bool useNchw = false; in execute() local
|
| D | RoiPooling.cpp | 140 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
157 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling()
203 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
|
| D | InstanceNormalization.cpp | 79 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
|
| D | Conv2D.cpp | 56 bool useNchw = false; member
267 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv()
381 int32_t dilationHeightFactor, int32_t activation, bool useNchw, in convQuant8PerChannel()
|
| D | RoiAlign.cpp | 314 int32_t heightSamplingRatio, int32_t widthSamplingRatio, bool useNchw, in roiAlign()
368 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
|
| D | GenerateProposals.cpp | 951 bool useNchw, std::vector<float>* scoresOutData, in generateProposalsFloat32Compute()
969 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat32()
1012 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat16()
1060 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsQuant()
1165 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
|
| D | Pooling.cpp | 41 bool useNchw = false; member
|
| D | TransposeConv2D.cpp | 54 bool useNchw = false; member
|
| /frameworks/ml/nn/common/include/ |
| D | CpuOperationUtils.h | 141 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout()
165 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout()
|