| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.cpp | 44 bool ModelBuilder::badState(const char* name) { in badState()
56 int ModelBuilder::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType()
63 int ModelBuilder::addOperand(const ANeuralNetworksOperandType& type) { in addOperand()
107 int ModelBuilder::setOperandValue(uint32_t index, const void* buffer, size_t length) { in setOperandValue()
175 int ModelBuilder::setOperandValueFromModel(uint32_t index, const ModelBuilder* value) { in setOperandValueFromModel()
204 int ModelBuilder::setOperandSymmPerChannelQuantParams( in setOperandSymmPerChannelQuantParams()
239 int ModelBuilder::setOperandExtensionData(uint32_t index, const void* data, size_t length) { in setOperandExtensionData()
278 int ModelBuilder::copyLargeValuesToSharedMemory() { in copyLargeValuesToSharedMemory()
311 int ModelBuilder::setOperandValueFromMemory(uint32_t index, const Memory* memory, uint32_t offset, in setOperandValueFromMemory()
349 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, in addOperation()
[all …]
|
| D | CompilationBuilder.h | 35 class ModelBuilder; variable
44 CompilationBuilder(const ModelBuilder* model,
64 const ModelBuilder* getModel() const { return mModel; } in getModel()
74 const ModelBuilder* mModel;
|
| D | NeuralNetworks.cpp | 714 const ModelBuilder* m = reinterpret_cast<const ModelBuilder*>(model); in ANeuralNetworksModel_getSupportedOperationsForDevices()
792 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksCompilation_createForDevices()
1038 ModelBuilder* m = new (std::nothrow) ModelBuilder(); in ANeuralNetworksModel_create()
1050 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_free()
1060 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_finish()
1071 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_addOperand()
1082 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValue()
1095 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValueFromMemory()
1106 const ModelBuilder* val = reinterpret_cast<const ModelBuilder*>(value); in ANeuralNetworksModel_setOperandValueFromModel()
1107 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValueFromModel()
[all …]
|
| D | ModelBuilder.h | 39 class ModelBuilder {
41 ModelBuilder() {} in ModelBuilder() function
49 int setOperandValueFromModel(uint32_t index, const ModelBuilder* value);
120 const ModelBuilder* getReferencedModel(uint32_t i) const { in getReferencedModel()
124 const ModelBuilder* getReferencedModel(const hal::Operand& operand) const { in getReferencedModel()
232 std::vector<const ModelBuilder*> mReferencedModels;
|
| D | ExecutionBuilder.h | 45 class ModelBuilder; variable
99 const ModelBuilder* getModel() const { return mModel; } in getModel()
100 const ModelBuilder* getSourceModel(uint32_t index) const;
143 const ModelBuilder* mModel;
229 StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model,
302 const ModelBuilder* mModel;
|
| D | ExecutionPlan.cpp | 71 int compile(const Device& device, const ModelBuilder& model, int executionPreference, in compile()
99 int copyOperandExtraParams(ModelBuilder& model, uint32_t toOperandIndex, in copyOperandExtraParams()
135 OperandTracker(const ModelBuilder* model, OperationReadyCallback cb);
142 const ModelBuilder* mModel;
147 OperandTracker::OperandTracker(const ModelBuilder* model, OperationReadyCallback cb) in OperandTracker()
211 const ModelBuilder& sourceModel = *getSourceModel(); in addOperand()
289 const ModelBuilder* model = sourceModel.getReferencedModel(operand); in addOperand()
427 const ModelBuilder* ExecutionStep::getSourceModel() const { in getSourceModel()
479 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel()
616 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish()
[all …]
|
| D | ExecutionPlan.h | 94 uint32_t addModel(const ModelBuilder* model) { in addModel()
100 const ModelBuilder* getModel(uint32_t index) const { return mModels[index]; } in getModel()
105 std::vector<const ModelBuilder*> mModels;
147 int finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize,
150 const ModelBuilder* getStepModel() const { return &mStepModel; } in getStepModel()
175 const ModelBuilder* getSourceModel() const;
184 ModelBuilder mStepModel; // An excerpt of a source model to be run by one device.
536 void becomeSingleStep(const std::shared_ptr<Device> device, const ModelBuilder* model);
644 SimpleBody(std::shared_ptr<Device> device, const ModelBuilder* model, in SimpleBody()
658 const ModelBuilder* mModel;
|
| D | CompilationBuilder.cpp | 41 CompilationBuilder::CompilationBuilder(const ModelBuilder* model, in CompilationBuilder()
|
| D | Android.bp | 58 "ModelBuilder.cpp",
|
| D | Memory.h | 45 class ModelBuilder; variable
|
| D | ExecutionBuilder.cpp | 101 const ModelBuilder* ExecutionBuilder::getSourceModel(uint32_t index) const { in getSourceModel()
978 StepExecutor::StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model, in StepExecutor()
|
| D | Memory.cpp | 350 const ModelBuilder* model = compilation.getModel(); in addRole()
|
| /frameworks/ml/nn/runtime/test/ |
| D | Bridge.h | 31 class ModelBuilder; variable
35 void graphDump(const char* name, const ModelBuilder* model, std::ostream* outStream = &std::cout);
|
| D | Bridge.cpp | 30 void graphDump(const char* name, const ModelBuilder* model, std::ostream* outStream) { in graphDump()
|
| D | TestPartitioningRandom.cpp | 99 using ModelBuilder = nn::ModelBuilder; typedef
192 const ModelBuilder* builder() const { in builder()
193 return reinterpret_cast<const ModelBuilder*>(getHandle()); in builder()
213 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in TestCompilation()
429 ModelStats(const ModelBuilder* model) : mBuilder(model) {} in ModelStats()
431 : mBuilder(reinterpret_cast<const ModelBuilder*>(model->getHandle())) {} in ModelStats()
445 const ModelBuilder* mBuilder;
|
| D | TestPartitioning.cpp | 159 using ModelBuilder = ::android::nn::ModelBuilder; typedef
292 void dump(const char* name, const ModelBuilder* model) { in dump()
799 return reinterpret_cast<ModelBuilder*>(getHandle()) in partitionTheWork()
807 const ModelBuilder* mb = reinterpret_cast<const ModelBuilder*>(getHandle()); in dump()
883 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in PartitioningCompilation()
1121 void buildDefinitionMap(const ModelBuilder* model, std::map<uint32_t, uint32_t>* defMap) { in buildDefinitionMap()
1235 bool compare(const ModelBuilder* modelA, const ModelBuilder* modelB, in compare()
1387 reinterpret_cast<const ModelBuilder*>(model->getHandle()), in compare()
|
| D | TestCompliance.cpp | 41 auto modelBuilder = reinterpret_cast<const ModelBuilder*>(wrapperModel.getHandle()); in createHidlModel()
|
| D | TestExecution.cpp | 571 nn::ModelBuilder* m = reinterpret_cast<nn::ModelBuilder*>(model->getHandle()); in TestCompilation()
|