| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Convolutions.cpp | 36 bool useNchw = false; in conv2DExplicitConstructor() local
88 bool useNchw = false; in conv2DImplicitConstructor() local
275 bool useNchw = false; in depthwiseConv2DExplicitConstructor() local
328 bool useNchw = false; in depthwiseConv2DImplicitConstructor() local
528 bool useNchw = op->inputs[11]->value<bool8>(); in groupedConv2DExplicitConstructor() local
580 bool useNchw = op->inputs[8]->value<bool8>(); in groupedConv2DImplicitConstructor() local
682 bool useNchw = op->inputs[10]->value<bool8>(); in transposeConv2DExplicitConstructor() local
726 bool useNchw = op->inputs[8]->value<bool8>(); in transposeConv2DImplicitConstructor() local
|
| D | BoundingBox.cpp | 42 bool useNchw; in roiConstructor() local
85 bool useNchw; in roiFinalizer() local
180 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 | Reshape.cpp | 29 bool useNchw = false; in spaceToDepthConstructor() local
84 bool useNchw = false; in depthToSpaceConstructor() local
172 bool useNchw = false; in batchToSpaceConstructor() local
230 bool useNchw = false; in spaceToBatchConstructor() local
|
| D | Resize.cpp | 27 bool useNchw = op->inputs.size() > 3 ? static_cast<bool>(op->inputs[3]->value<bool8>()) : false; in resizeOpConstructor() local
|
| /frameworks/ml/nn/common/operations/ |
| D | RoiPooling.cpp | 146 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
163 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling()
178 float widthStride, bool useNchw, int8_t* outputData, in roiPooling()
230 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
|
| D | ResizeImageOps.cpp | 150 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp()
228 const bool useNchw = getOptionalScalar(context, kLayoutScalar); in prepare() local
279 const bool useNchw = getOptionalScalar(context, kLayoutScalar); in execute() local
|
| D | InstanceNormalization.cpp | 92 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
|
| D | RoiAlign.cpp | 318 int32_t heightSamplingRatio, int32_t widthSamplingRatio, bool useNchw, in roiAlign()
385 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
|
| D | Conv2D.cpp | 69 bool useNchw = false; member
311 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv()
488 int32_t dilationHeightFactor, int32_t activation, bool useNchw, in convQuant8PerChannel()
|
| D | GenerateProposals.cpp | 1060 bool useNchw, std::vector<float>* scoresOutData, in generateProposalsFloat32Compute()
1078 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat32()
1121 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat16()
1170 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsQuant()
1281 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
|
| D | Pooling.cpp | 49 bool useNchw = false; member
|
| D | TransposeConv2D.cpp | 64 bool useNchw = false; member
|
| /frameworks/ml/nn/common/include/ |
| D | CpuOperationUtils.h | 159 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout()
183 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout()
|