| /packages/modules/NeuralNetworks/runtime/ |
| D | ModelBuilder.cpp | 59 bool ModelBuilder::badState(const char* name) { in badState()
71 int ModelBuilder::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType()
78 int ModelBuilder::addOperand(const ANeuralNetworksOperandType& type) { in addOperand()
128 int ModelBuilder::setOperandValue(uint32_t index, const void* buffer, size_t length) { in setOperandValue()
192 int ModelBuilder::setOperandValueFromModel(uint32_t index, const ModelBuilder* value) { in setOperandValueFromModel()
222 int ModelBuilder::setOperandSymmPerChannelQuantParams( in setOperandSymmPerChannelQuantParams()
260 int ModelBuilder::setOperandExtensionData(uint32_t index, const void* data, size_t length) { in setOperandExtensionData()
290 int ModelBuilder::copyLargeValuesToSharedMemory() { in copyLargeValuesToSharedMemory()
324 int ModelBuilder::setOperandValueFromMemory(uint32_t index, const RuntimeMemory* memory, in setOperandValueFromMemory()
362 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, in addOperation()
[all …]
|
| D | NeuralNetworks.cpp | 779 const ModelBuilder* m = reinterpret_cast<const ModelBuilder*>(model); in ANeuralNetworksModel_getSupportedOperationsForDevices()
857 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksCompilation_createForDevices()
1115 ModelBuilder* m = new (std::nothrow) ModelBuilder(); in ANeuralNetworksModel_create()
1127 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_free()
1137 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_finish()
1148 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_addOperand()
1159 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValue()
1172 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValueFromMemory()
1183 const ModelBuilder* val = reinterpret_cast<const ModelBuilder*>(value); in ANeuralNetworksModel_setOperandValueFromModel()
1184 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValueFromModel()
[all …]
|
| D | ModelBuilder.h | 40 class ModelBuilder {
42 ModelBuilder() {} in ModelBuilder() function
50 int setOperandValueFromModel(uint32_t index, const ModelBuilder* value);
122 const ModelBuilder* getReferencedModel(uint32_t i) const { in getReferencedModel()
126 const ModelBuilder* getReferencedModel(const Operand& operand) const { in getReferencedModel()
246 std::vector<const ModelBuilder*> mReferencedModels;
|
| D | CompilationBuilder.h | 41 class ModelBuilder; variable
50 CompilationBuilder(const ModelBuilder* model,
80 const ModelBuilder* getModel() const { return mModel; } in getModel()
106 const ModelBuilder* mModel;
|
| D | ExecutionBuilder.h | 50 class ModelBuilder; variable
117 const ModelBuilder* getModel() const { return mModel; } in getModel()
118 const ModelBuilder* getSourceModel(uint32_t index) const;
188 const ModelBuilder* mModel;
344 StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model,
449 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);
849 SimpleBody(std::shared_ptr<Device> device, const ModelBuilder* model, in SimpleBody()
865 const ModelBuilder* mModel;
|
| D | ExecutionPlan.cpp | 105 int compile(const Device& device, const ModelBuilder& model, int executionPreference, in compile()
139 int copyOperandExtraParams(ModelBuilder& model, uint32_t toOperandIndex, in copyOperandExtraParams()
172 OperandTracker(const ModelBuilder* model, OperationReadyCallback cb);
179 const ModelBuilder* mModel;
184 OperandTracker::OperandTracker(const ModelBuilder* model, OperationReadyCallback cb) in OperandTracker()
470 const ModelBuilder& sourceModel = *getSourceModel(); in addOperand()
540 const ModelBuilder* model = sourceModel.getReferencedModel(operand); in addOperand()
732 const ModelBuilder* ExecutionStep::getSourceModel() const { in getSourceModel()
784 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel()
923 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish()
[all …]
|
| D | Telemetry.cpp | 105 DataClass evalInputDataClass(const ModelBuilder* m) { in evalInputDataClass()
114 DataClass evalOutputDataClass(const ModelBuilder* m) { in evalOutputDataClass()
|
| D | FlatbufferModelBuilder.h | 32 class FlatbufferModelBuilder : public ModelBuilder {
|
| D | Android.bp | 98 "ModelBuilder.cpp",
294 "ModelBuilder.cpp",
|
| D | Memory.h | 47 class ModelBuilder; variable
|
| D | ExecutionBuilder.cpp | 55 static bool validateOutputShapesFromDriver(ErrorStatus executionStatus, const ModelBuilder* model, in validateOutputShapesFromDriver()
107 static bool validateOutputShapesFromDriver(int executionResultCode, const ModelBuilder* model, in validateOutputShapesFromDriver()
174 const ModelBuilder* ExecutionBuilder::getSourceModel(uint32_t index) const { in getSourceModel()
1360 StepExecutor::StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model, in StepExecutor()
|
| D | CompilationBuilder.cpp | 43 CompilationBuilder::CompilationBuilder(const ModelBuilder* model, in CompilationBuilder()
|
| D | Memory.cpp | 343 const ModelBuilder* model = compilation.getModel(); in addRole()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestPartitioningRandom.cpp | 112 using ModelBuilder = nn::ModelBuilder; typedef
205 const ModelBuilder* builder() const { in builder()
206 return reinterpret_cast<const ModelBuilder*>(getHandle()); in builder()
226 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in TestCompilation()
442 ModelStats(const ModelBuilder* model) : mBuilder(model) {} in ModelStats()
444 : mBuilder(reinterpret_cast<const ModelBuilder*>(model->getHandle())) {} in ModelStats()
458 const ModelBuilder* mBuilder;
|
| D | TestPartitioning.cpp | 169 using ModelBuilder = ::android::nn::ModelBuilder; typedef
303 void dump(const char* name, const ModelBuilder* model) { in dump()
850 return reinterpret_cast<ModelBuilder*>(getHandle()) in partitionTheWork()
858 const ModelBuilder* mb = reinterpret_cast<const ModelBuilder*>(getHandle()); in dump()
944 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in PartitioningCompilation()
1256 void buildDefinitionMap(const ModelBuilder* model, std::map<uint32_t, uint32_t>* defMap) { in buildDefinitionMap()
1369 bool compare(const ModelBuilder* modelA, const ModelBuilder* modelB, in compare()
1533 reinterpret_cast<const ModelBuilder*>(model->getHandle()), in compare()
|
| D | TestCompliance.cpp | 45 auto modelBuilder = reinterpret_cast<const ModelBuilder*>(wrapperModel.getHandle()); in testAvailableSinceVersion()
|
| D | TestExecution.cpp | 607 nn::ModelBuilder* m = reinterpret_cast<nn::ModelBuilder*>(model->getHandle()); in TestCompilation()
|