Home
last modified time | relevance | path

Searched defs:useNchw (Results 1 – 15 of 15) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DConvolutions.cpp36 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
DBoundingBox.cpp44 bool useNchw; in roiConstructor() local
89 bool useNchw; in roiFinalizer() local
185 bool useNchw = op->inputs[2]->value<bool8>(); in heatmapMaxKeypointConstructor() local
DPoolings.cpp36 bool useNchw = false; in poolingExplicitOpConstructor() local
75 bool useNchw = false; in poolingImplicitOpConstructor() local
DResize.cpp27 bool useNchw = op->inputs.size() > 3 ? static_cast<bool>(op->inputs[3]->value<bool8>()) : false; in resizeOpConstructor() local
DReshape.cpp29 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
/packages/modules/NeuralNetworks/common/operations/
DRoiPooling.cpp147 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
164 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling()
179 float widthStride, bool useNchw, int8_t* outputData, in roiPooling()
231 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
DResizeImageOps.cpp152 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp()
234 const bool useNchw = getOptionalScalar(context, kLayoutScalar); in prepare() local
285 const bool useNchw = getOptionalScalar(context, kLayoutScalar); in execute() local
DInstanceNormalization.cpp93 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
DDepthwiseConv2D.cpp56 bool useNchw = false; member
378 int32_t depthMultiplier, int32_t activation, bool useNchw, T_Input* outputData, in depthwiseConv()
401 int32_t depthMultiplier, int32_t activation, bool useNchw, in depthwiseConvQuant8PerChannel()
DRoiAlign.cpp320 int32_t heightSamplingRatio, int32_t widthSamplingRatio, bool useNchw, in roiAlign()
388 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
DConv2D.cpp69 bool useNchw = false; member
337 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv()
514 int32_t dilationHeightFactor, int32_t activation, bool useNchw, in convQuant8PerChannel()
DGenerateProposals.cpp1067 bool useNchw, std::vector<float>* scoresOutData, in generateProposalsFloat32Compute()
1085 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat32()
1128 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat16()
1177 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsQuant()
1290 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
DPooling.cpp51 bool useNchw = false; member
DTransposeConv2D.cpp64 bool useNchw = false; member
/packages/modules/NeuralNetworks/common/include/
DCpuOperationUtils.h160 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout()
184 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout()