Home
last modified time | relevance | path

Searched refs:dimensionCount (Results 1 – 14 of 14) sorted by relevance

/frameworks/ml/nn/runtime/test/
DTestValidateModel.cpp34 operand_0.dimensionCount = 0; in TEST_F()
40 operand_1.dimensionCount = 0; in TEST_F()
46 operand_2.dimensionCount = 4; in TEST_F()
54 operand_3.dimensionCount = 2; in TEST_F()
62 operand_4.dimensionCount = 2; in TEST_F()
70 operand_5.dimensionCount = 2; in TEST_F()
78 operand_6.dimensionCount = 1; in TEST_F()
86 operand_7.dimensionCount = 4; in TEST_F()
94 operand_8.dimensionCount = 4; in TEST_F()
102 operand_9.dimensionCount = 4; in TEST_F()
[all …]
DTestValidateOperations.cpp46 .dimensionCount = dimCount, in getOpType()
333 .dimensionCount = 0, in argMinMaxTest()
339 .dimensionCount = 3, in argMinMaxTest()
388 .dimensionCount = 0, in expandDimsTest()
410 .dimensionCount = 0, in gatherTest()
415 .dimensionCount = 4, in gatherTest()
434 .type = operandCode, .dimensionCount = 4, .dimensions = inputDimensions}; in quantizeOpTest()
436 .dimensionCount = 4, in quantizeOpTest()
458 .dimensionCount = 1, in TEST()
465 .dimensionCount = 2, in TEST()
[all …]
DTestOperandExtraParams.cpp53 .dimensionCount = 0, in createOperand()
64 .dimensionCount = 4, in createOperand()
70 .dimensionCount = 4, in createOperand()
76 .dimensionCount = 4, in createOperand()
82 .dimensionCount = 4, in createOperand()
88 .dimensionCount = 4, in createOperand()
DTestValidation.cpp53 .type = type, .dimensionCount = 0, .dimensions = nullptr}; in addScalarOperand()
67 .dimensionCount = sizeof(dimensions) / sizeof(dimensions[0]), in addTensorOperand()
97 .dimensionCount = 1,
103 .dimensionCount = 0,
150 .dimensionCount = 1, in SetUp()
153 .type = ANEURALNETWORKS_INT32, .dimensionCount = 0, .dimensions = nullptr}; in SetUp()
221 .type = ANEURALNETWORKS_FLOAT32, .dimensionCount = 0, .dimensions = nullptr}; in TEST_F()
228 .dimensionCount = 0, in TEST_F()
239 .dimensionCount = 0, in TEST_F()
252 .dimensionCount = 1, in TEST_F()
[all …]
/frameworks/ml/nn/common/
DUtilsTest.cpp78 .dimensionCount = 0, in TEST()
DUtils.cpp305 return tensorHasUnspecifiedDimensions(type->type, type->dimensions, type->dimensionCount); in tensorHasUnspecifiedDimensions()
383 NN_RET_CHECK_EQ(type.dimensionCount, 0u) << tag << " invalid dimensions for scalar type"; in validateScalarDimensions()
425 NN_RET_CHECK_GT(type.dimensionCount, 0u) << tag << " invalid operand dimensions"; in validateTensorDimensions()
426 for (uint32_t i = 0; i < type.dimensionCount; i++) { in validateTensorDimensions()
436 NN_RET_CHECK_EQ(type.dimensionCount == 0, type.dimensions == nullptr); in validateOperandTypeHelper()
/frameworks/ml/nn/extensions/
DREADME.md120 .dimensionCount = dimensionCount,
/frameworks/ml/nn/runtime/include/
DNeuralNetworksWrapperExtensions.h76 .dimensionCount = static_cast<uint32_t>(dimensions.size()), in dimensions()
DNeuralNetworksWrapper.h123 .dimensionCount = static_cast<uint32_t>(dimensions.size()), in dimensions()
135 .dimensionCount = static_cast<uint32_t>(dimensions.size()), in OperandType()
DNeuralNetworks.h5279 uint32_t dimensionCount; member
/frameworks/ml/nn/runtime/
DExecutionBuilder.cpp62 if (operand.dimensions.size() != newType->dimensionCount) { in checkDimensionInfo()
66 for (uint32_t i = 0; i < newType->dimensionCount; i++) { in checkDimensionInfo()
155 const uint32_t count = newType->dimensionCount; in updateDimensionInfo()
DModelBuilder.cpp87 hidl_vec<uint32_t>(type.dimensions, type.dimensions + type.dimensionCount), in addOperand()
DNeuralNetworks.cpp518 static_assert(offsetof(ANeuralNetworksOperandType, dimensionCount) == 4,
DExecutionPlan.cpp289 .dimensionCount = static_cast<uint32_t>(operand.dimensions.size()), in addOperand()