| /packages/modules/NeuralNetworks/runtime/ |
| D | ModelBuilder.cpp | 55 bool ModelBuilder::badState(const char* name) { in badState()
67 int ModelBuilder::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType()
74 int ModelBuilder::addOperand(const ANeuralNetworksOperandType& type) { in addOperand()
123 int ModelBuilder::setOperandValue(uint32_t index, const void* buffer, size_t length) { in setOperandValue()
187 int ModelBuilder::setOperandValueFromModel(uint32_t index, const ModelBuilder* value) { in setOperandValueFromModel()
217 int ModelBuilder::setOperandSymmPerChannelQuantParams( in setOperandSymmPerChannelQuantParams()
255 int ModelBuilder::setOperandExtensionData(uint32_t index, const void* data, size_t length) { in setOperandExtensionData()
285 int ModelBuilder::copyLargeValuesToSharedMemory() { in copyLargeValuesToSharedMemory()
318 int ModelBuilder::setOperandValueFromMemory(uint32_t index, const RuntimeMemory* memory, in setOperandValueFromMemory()
356 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, in addOperation()
[all …]
|
| D | NeuralNetworks.cpp | 739 const ModelBuilder* m = reinterpret_cast<const ModelBuilder*>(model); in ANeuralNetworksModel_getSupportedOperationsForDevices()
817 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksCompilation_createForDevices()
1063 ModelBuilder* m = new (std::nothrow) ModelBuilder(); in ANeuralNetworksModel_create()
1075 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_free()
1085 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_finish()
1096 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_addOperand()
1107 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValue()
1120 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValueFromMemory()
1131 const ModelBuilder* val = reinterpret_cast<const ModelBuilder*>(value); in ANeuralNetworksModel_setOperandValueFromModel()
1132 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 Operand& operand) const { in getReferencedModel()
232 std::vector<const ModelBuilder*> mReferencedModels;
|
| D | CompilationBuilder.h | 36 class ModelBuilder; variable
45 CompilationBuilder(const ModelBuilder* model,
72 const ModelBuilder* getModel() const { return mModel; } in getModel()
89 const ModelBuilder* mModel;
|
| D | ExecutionPlan.cpp | 70 int compile(const Device& device, const ModelBuilder& model, int executionPreference, in compile()
99 int copyOperandExtraParams(ModelBuilder& model, uint32_t toOperandIndex, in copyOperandExtraParams()
132 OperandTracker(const ModelBuilder* model, OperationReadyCallback cb);
139 const ModelBuilder* mModel;
144 OperandTracker::OperandTracker(const ModelBuilder* model, OperationReadyCallback cb) in OperandTracker()
430 const ModelBuilder& sourceModel = *getSourceModel(); in addOperand()
500 const ModelBuilder* model = sourceModel.getReferencedModel(operand); in addOperand()
692 const ModelBuilder* ExecutionStep::getSourceModel() const { in getSourceModel()
744 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel()
880 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish()
[all …]
|
| D | ExecutionBuilder.h | 49 class ModelBuilder; variable
110 const ModelBuilder* getModel() const { return mModel; } in getModel()
111 const ModelBuilder* getSourceModel(uint32_t index) const;
169 const ModelBuilder* mModel;
322 StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model,
427 const ModelBuilder* mModel;
|
| D | ExecutionPlan.h | 104 uint32_t addModel(const ModelBuilder* model) { in addModel()
110 const ModelBuilder* getModel(uint32_t index) const { return mModels[index]; } in getModel()
115 std::vector<const ModelBuilder*> mModels;
277 int finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize,
280 const ModelBuilder* getStepModel() const { return &mStepModel; } in getStepModel()
318 const ModelBuilder* getSourceModel() const;
327 ModelBuilder mStepModel; // An excerpt of a source model to be run by one device.
712 void becomeSingleStep(const std::shared_ptr<Device> device, const ModelBuilder* model);
848 SimpleBody(std::shared_ptr<Device> device, const ModelBuilder* model, in SimpleBody()
863 const ModelBuilder* mModel;
|
| D | Android.bp | 94 "ModelBuilder.cpp",
215 "ModelBuilder.cpp",
|
| D | ExecutionBuilder.cpp | 54 static bool validateOutputShapesFromDriver(ErrorStatus executionStatus, const ModelBuilder* model, in validateOutputShapesFromDriver()
106 static bool validateOutputShapesFromDriver(int executionResultCode, const ModelBuilder* model, in validateOutputShapesFromDriver()
173 const ModelBuilder* ExecutionBuilder::getSourceModel(uint32_t index) const { in getSourceModel()
1313 StepExecutor::StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model, in StepExecutor()
|
| D | Memory.h | 47 class ModelBuilder; variable
|
| D | CompilationBuilder.cpp | 42 CompilationBuilder::CompilationBuilder(const ModelBuilder* model, in CompilationBuilder()
|
| D | Memory.cpp | 335 const ModelBuilder* model = compilation.getModel(); in addRole()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestPartitioningRandom.cpp | 115 using ModelBuilder = nn::ModelBuilder; typedef
208 const ModelBuilder* builder() const { in builder()
209 return reinterpret_cast<const ModelBuilder*>(getHandle()); in builder()
229 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in TestCompilation()
445 ModelStats(const ModelBuilder* model) : mBuilder(model) {} in ModelStats()
447 : mBuilder(reinterpret_cast<const ModelBuilder*>(model->getHandle())) {} in ModelStats()
461 const ModelBuilder* mBuilder;
|
| D | TestPartitioning.cpp | 168 using ModelBuilder = ::android::nn::ModelBuilder; typedef
302 void dump(const char* name, const ModelBuilder* model) { in dump()
849 return reinterpret_cast<ModelBuilder*>(getHandle()) in partitionTheWork()
857 const ModelBuilder* mb = reinterpret_cast<const ModelBuilder*>(getHandle()); in dump()
943 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in PartitioningCompilation()
1255 void buildDefinitionMap(const ModelBuilder* model, std::map<uint32_t, uint32_t>* defMap) { in buildDefinitionMap()
1368 bool compare(const ModelBuilder* modelA, const ModelBuilder* modelB, in compare()
1532 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 | 606 nn::ModelBuilder* m = reinterpret_cast<nn::ModelBuilder*>(model->getHandle()); in TestCompilation()
|