Home
last modified time | relevance | path

Searched defs:keepDims (Results 1 – 24 of 24) sorted by relevance

/frameworks/ml/nn/runtime/test/generated/models/
Dmean.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
39 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_float16.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
39 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_float_1_relaxed.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
41 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_quant8_1.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
39 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_quant8_2.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
39 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_float_2.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
39 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_float_1.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
39 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_float_2_relaxed.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
41 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dmean_relaxed.model.cpp11 auto keepDims = model->addOperand(&type2); in CreateModel() local
41 auto keepDims = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
/frameworks/ml/nn/common/operations/
DSimpleMath.cpp52 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
67 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
DReduce.cpp123 bool keepDims = context->getInputValue<bool8>(kInputKeepDims); in prepare() local
/frameworks/ml/nn/runtime/test/specs/V1_1/
Dmean_float_2.mod.py4 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_float_1.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_quant8_2.mod.py4 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_quant8_1.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
Dmean.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_float_1_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_float_2_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 0) variable
/frameworks/ml/nn/runtime/test/specs/V1_2/
Dmean_float16.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
/frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/
DReduce.cpp40 bool keepDims; in reduceOpConstructor() local
/frameworks/ml/nn/common/
DOperationsUtils.cpp697 bool keepDims, in meanPrepare()
DCpuExecutor.cpp1403 int32_t keepDims = getScalarData<int32_t>(mOperands[ins[2]]); in executeOperation() local
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp901 ANeuralNetworksOperandType keepDims = getOpType(ANEURALNETWORKS_INT32); in meanOpTest() local