| /frameworks/ml/nn/runtime/test/ |
| D | TestPartitioningRandom.cpp | 402 const uint32_t operandIndex = model->addOperand(&biasType); in makeRnnSpecialInput() local
405 model->setOperandValue(operandIndex, biasValue); in makeRnnSpecialInput()
406 return int(operandIndex); in makeRnnSpecialInput()
420 const uint32_t operandIndex = model->addOperand(&permType); in makeTransposeSpecialInput() local
422 model->setOperandValue(operandIndex, permValue); in makeTransposeSpecialInput()
423 return int(operandIndex); in makeTransposeSpecialInput()
827 const uint32_t operandIndex = model.addOperand(&activationFunctionType); in TEST_P() local
833 model.setOperandValue(operandIndex, activationFunction); in TEST_P()
834 operationInputs[operationInputIndex] = operandIndex; in TEST_P()
838 const int operandIndex = (this->*(operationPattern.mMakeSpecialInput))( in TEST_P() local
[all …]
|
| D | TestOperandExtraParams.cpp | 120 int operandIndex = nextOperandIndex++; in testAddingWithSymmPerChannelQuantParams() local
121 EXPECT_EQ(ANeuralNetworksModel_setOperandSymmPerChannelQuantParams(mModel, operandIndex, in testAddingWithSymmPerChannelQuantParams()
|
| D | TestValidation.cpp | 444 const int32_t operandIndex = addTensorOperand(ANEURALNETWORKS_TENSOR_QUANT8_SYMM_PER_CHANNEL); in TEST_F() local
453 EXPECT_EQ(ANeuralNetworksModel_setOperandSymmPerChannelQuantParams(nullptr, operandIndex, in TEST_F()
457 ANeuralNetworksModel_setOperandSymmPerChannelQuantParams(mModel, operandIndex, nullptr), in TEST_F()
459 EXPECT_EQ(ANeuralNetworksModel_setOperandSymmPerChannelQuantParams(mModel, operandIndex + 1, in TEST_F()
462 EXPECT_EQ(ANeuralNetworksModel_setOperandSymmPerChannelQuantParams(mModel, operandIndex, in TEST_F()
474 const int32_t operandIndex = in TEST_F() local
484 EXPECT_EQ(ANeuralNetworksModel_setOperandSymmPerChannelQuantParams(mModel, operandIndex, in TEST_F()
490 const int32_t operandIndex = in TEST_F() local
495 ANeuralNetworksModel_setOperandExtensionData(nullptr, operandIndex, &data, dataLength), in TEST_F()
498 ANeuralNetworksModel_setOperandExtensionData(mModel, operandIndex, nullptr, dataLength), in TEST_F()
[all …]
|
| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.cpp | 169 mLargeOperandValues.push_back(LargeValue{.operandIndex = index, .buffer = buffer}); in setOperandValue()
285 Operand& operand = mOperands[l.operandIndex]; in copyLargeValuesToSharedMemory()
303 Operand& operand = mOperands[l.operandIndex]; in copyLargeValuesToSharedMemory()
443 const uint32_t operandIndex = indexList[i]; in identifyInputsAndOutputs() local
444 if (operandIndex >= mOperands.size()) { in identifyInputsAndOutputs()
448 << operandIndex << " as this exceeds the numbe of operands " in identifyInputsAndOutputs()
452 (*indexVector)[i] = operandIndex; in identifyInputsAndOutputs()
453 Operand& operand = mOperands[operandIndex]; in identifyInputsAndOutputs()
456 << operandIndex in identifyInputsAndOutputs()
819 for (uint32_t operandIndex : mOperations[operationIndex].inputs) { in sortIntoRunOrder() local
[all …]
|
| D | ExecutionPlan.cpp | 154 for (uint32_t operandIndex : operation.inputs) { in OperandTracker() local
155 auto lifetime = mModel->getOperand(operandIndex).lifetime; in OperandTracker()
159 mOperandToOperations.emplace(operandIndex, operationIndex); in OperandTracker()
172 for (uint32_t operandIndex : operation.outputs) { in markProcessed() local
173 auto range = mOperandToOperations.equal_range(operandIndex); in markProcessed()
1048 std::shared_ptr<Controller> controller, SourceOperandIndex operandIndex) const { in getBuffer()
1053 if (auto it = sourceOperandToOffsetOfTemporary.find(operandIndex); in getBuffer()
1058 } else if (auto it = sourceOperandToInputIndex.find(operandIndex); in getBuffer()
1062 } else if (auto it = sourceOperandToOutputIndex.find(operandIndex); in getBuffer()
1066 } else if (auto it = sourceOperandToConstantReference.find(operandIndex); in getBuffer()
[all …]
|
| D | ModelBuilder.h | 209 uint32_t operandIndex; member
|
| D | ExecutionPlan.h | 606 SourceOperandIndex operandIndex) const;
610 int readConditionValue(std::shared_ptr<Controller> controller, SourceOperandIndex operandIndex,
|
| D | Manager.cpp | 219 for (uint32_t operandIndex : operands) { in getSupportedOperations() local
220 const Operand& operand = hidlModel.main.operands[operandIndex]; in getSupportedOperations()
|
| /frameworks/ml/nn/tools/test_generator/test_harness/include/ |
| D | TestHarness.h | 372 for (const auto operandIndex : operation.inputs) { in hasQuant8CoupledOperands()
373 if (subgraph.operands[operandIndex].type == in hasQuant8CoupledOperands()
379 for (const auto operandIndex : operation.outputs) { in hasQuant8CoupledOperands()
380 if (subgraph.operands[operandIndex].type == in hasQuant8CoupledOperands()
|
| /frameworks/ml/nn/extensions/ |
| D | README.md | 140 CHECK_EQ(ANeuralNetworksModel_setOperandExtensionData(model, operandIndex, ¶ms, sizeof(params)),
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 746 const uint32_t operandIndex = indexes[i]; in updateForArguments() local
748 RunTimeOperandInfo& to = operands[operandIndex]; in updateForArguments()
1913 const uint32_t operandIndex = outputIndexes[i]; in setOutputShapes() local
1914 const RunTimeOperandInfo& from = operands[operandIndex]; in setOutputShapes()
|
| D | ValidateHal.cpp | 762 const uint32_t operandIndex = operandIndexes[requestArgumentIndex]; in validateRequestArguments() local
763 const Operand& operand = operands[operandIndex]; in validateRequestArguments()
|