Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestValidateModel.cpp35 operand_0.dimensionCount = 0; in TEST_F()
41 operand_1.dimensionCount = 0; in TEST_F()
47 operand_2.dimensionCount = 4; in TEST_F()
55 operand_3.dimensionCount = 2; in TEST_F()
63 operand_4.dimensionCount = 2; in TEST_F()
71 operand_5.dimensionCount = 2; in TEST_F()
79 operand_6.dimensionCount = 1; in TEST_F()
87 operand_7.dimensionCount = 4; in TEST_F()
95 operand_8.dimensionCount = 4; in TEST_F()
103 operand_9.dimensionCount = 4; in TEST_F()
[all …]
DTestValidateOperations.cpp59 .dimensionCount = dimCount, in getOpType()
85 operandType.dimensionCount != that.operandType.dimensionCount) { in operator ==()
105 operandType.dimensions + operandType.dimensionCount, in operator ==()
132 if (operandType.dimensionCount < that.operandType.dimensionCount) return true; in operator <()
188 assert(op.operandType.dimensionCount == *mRangeMin); in MutationsWithValidRank()
224 if (operandTypeWithParams.operandType.dimensionCount == newRank) { in ModifyForRank()
237 std::min(operandTypeWithParams.operandType.dimensionCount, newRank); in ModifyForRank()
245 .dimensionCount = newRank, in ModifyForRank()
665 << "dimensionCount: " << operandType.dimensionCount << ", dimensions: ["; in operator <<()
666 std::for_each(operandType.dimensions, operandType.dimensions + operandType.dimensionCount, in operator <<()
[all …]
DTestOperandExtraParams.cpp55 .dimensionCount = 0, in createOperand()
66 .dimensionCount = 4, in createOperand()
72 .dimensionCount = 4, in createOperand()
78 .dimensionCount = 4, in createOperand()
84 .dimensionCount = 4, in createOperand()
90 .dimensionCount = 4, in createOperand()
96 .dimensionCount = 4, in createOperand()
DTestValidation.cpp63 .type = type, .dimensionCount = 0, .dimensions = nullptr}; in addScalarOperand()
80 .dimensionCount = static_cast<uint32_t>(dimensions.size()), in addTensorOperand()
119 .dimensionCount = 1,
125 .dimensionCount = 0,
172 .dimensionCount = 1, in SetUp()
175 .type = ANEURALNETWORKS_INT32, .dimensionCount = 0, .dimensions = nullptr}; in SetUp()
312 .type = ANEURALNETWORKS_INT32, .dimensionCount = 0, .dimensions = nullptr}; in addScalarOperand()
320 .dimensionCount = 1, in addTensorOperand()
396 .type = ANEURALNETWORKS_FLOAT32, .dimensionCount = 0, .dimensions = nullptr}; in TEST_F()
403 .dimensionCount = 0, in TEST_F()
[all …]
DTestFree.cpp36 .type = ANEURALNETWORKS_TENSOR_FLOAT32, .dimensionCount = 1, .dimensions = dimensions}; in createUnfinishedModel()
/frameworks/ml/nn/common/
DUtilsTest.cpp81 .dimensionCount = 0, in TEST()
103 .dimensionCount = std::size(dimensions), in TEST()
120 .dimensionCount = std::size(dimensions), in TEST()
DUtils.cpp421 return tensorHasUnspecifiedDimensions(type->type, type->dimensions, type->dimensionCount); in tensorHasUnspecifiedDimensions()
512 NN_RET_CHECK_EQ(type.dimensionCount, 0u) << tag << " invalid dimensions for scalar type"; in validateScalarDimensions()
562 NN_RET_CHECK_GT(type.dimensionCount, 0u) << tag << " invalid operand dimensions"; in validateTensorDimensions()
569 for (uint32_t i = 0; i < type.dimensionCount; i++) { in validateTensorDimensions()
585 NN_RET_CHECK_EQ(type.dimensionCount == 0, type.dimensions == nullptr); in validateOperandTypeHelper()
/frameworks/ml/nn/runtime/
DModelArgumentInfo.cpp100 const uint32_t count = newType->dimensionCount; in updateDimensionInfo()
DMemory.cpp113 NN_RET_CHECK(isTensor || type->dimensionCount == 0) in validate()
116 type->dimensions + type->dimensionCount); in validate()
DExecutionBuilder.cpp70 if (operand.dimensions.size() != newType->dimensionCount) { in checkDimensionInfo()
74 for (uint32_t i = 0; i < newType->dimensionCount; i++) { in checkDimensionInfo()
DModelBuilder.cpp95 hidl_vec<uint32_t>(type.dimensions, type.dimensions + type.dimensionCount), in addOperand()
DNeuralNetworks.cpp572 static_assert(offsetof(ANeuralNetworksOperandType, dimensionCount) == 4,
DExecutionPlan.cpp215 .dimensionCount = static_cast<uint32_t>(operand.dimensions.size()), in addOperand()
/frameworks/ml/nn/runtime/include/
DNeuralNetworksWrapperExtensions.h76 .dimensionCount = static_cast<uint32_t>(dimensions.size()), in dimensions()
DNeuralNetworksWrapper.h136 .dimensionCount = static_cast<uint32_t>(dimensions.size()), in dimensions()
149 .dimensionCount = static_cast<uint32_t>(dimensions.size()), in OperandType()
DNeuralNetworks.h6224 uint32_t dimensionCount; member
/frameworks/ml/nn/extensions/
DREADME.md125 .dimensionCount = dimensionCount,
/frameworks/ml/nn/tools/api/
DNeuralNetworks.t630 * unspecified dimensions is represented by setting dimensionCount to
636 * represented by setting dimensionCount to 0 and dimensions to NULL (just as if
650 uint32_t dimensionCount;