Searched refs:hidlModel (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | Manager.cpp | 64 void Device::getSupportedOperations(const Model& hidlModel, in getSupportedOperations() argument
69 std::tie(status, supportedOperations) = mInterface.getSupportedOperations(hidlModel); in getSupportedOperations()
74 outSupportedOperations->resize(hidlModel.operations.size()); in getSupportedOperations()
78 if (supportedOperations.size() != hidlModel.operations.size()) { in getSupportedOperations()
81 << hidlModel.operations.size(); in getSupportedOperations()
83 outSupportedOperations->resize(hidlModel.operations.size()); in getSupportedOperations()
103 const Operation &operation = hidlModel.operations[operationIndex]; in getSupportedOperations()
105 auto accumulateOperands = [&hidlModel, &accumulator](const hidl_vec<uint32_t>& operands) { in getSupportedOperations()
107 const Operand& operand = hidlModel.operands[operandIndex]; in getSupportedOperations()
|
| D | Manager.h | 43 void getSupportedOperations(const Model& hidlModel, hidl_vec<bool>* supportedOperations);
|
| D | ExecutionPlan.cpp | 45 Model hidlModel; in compile() local
46 model->setHidlModel(&hidlModel); in compile()
50 hidlModel, static_cast<ExecutionPreference>(executionPreference), preparedModelCallback); in compile()
918 Model hidlModel; in initialize() local
919 model->setHidlModel(&hidlModel); in initialize()
920 device->getSupportedOperations(hidlModel, &mSupportsOperationByIndex); in initialize()
|
| D | ExecutionBuilder.cpp | 389 Model hidlModel; in startCompute() local
390 mModel->setHidlModel(&hidlModel); in startCompute()
395 device->getSupportedOperations(hidlModel, &supports); in startCompute()
|
| /frameworks/ml/nn/runtime/test/ |
| D | Bridge.cpp | 31 Model hidlModel; in graphDump() local
32 model->setHidlModel(&hidlModel); in graphDump()
33 ::android::nn::graphDump(name, hidlModel, outStream); in graphDump()
|
| D | TestPartitioning.cpp | 195 HidlModel hidlModel; in dump() local
196 model->setHidlModel(&hidlModel); in dump()
197 std::cout << name << ": " << toString(hidlModel) << std::endl; in dump()
198 std::cout << "inputs: " << toString(hidlModel.inputIndexes) << std::endl; in dump()
199 std::cout << "outputs: " << toString(hidlModel.outputIndexes) << std::endl; in dump()
200 for (size_t i = 0, e = hidlModel.operations.size(); i < e; i++) { in dump()
201 std::cout << "operation[" << i << "]: " << toString(hidlModel.operations[i]) << std::endl; in dump()
214 HidlModel hidlModel; in graphDump() local
215 reinterpret_cast<const ModelBuilder*>(model.getHandle())->setHidlModel(&hidlModel); in graphDump()
216 hidlGraphDump(name, hidlModel); in graphDump()
|