| /frameworks/ml/nn/runtime/ |
| D | ExecutionPlan.h | 55 int addOperation(int operationIndex, const ModelBuilder& fromModel);
57 const ModelBuilder& fromModel, OperandKind kind);
91 int finishSubModel(const ModelBuilder* fromModel, bool* hasOutputOfUnknownSize,
94 const ModelBuilder* getSubModel() const { return &mSubModel; } in getSubModel()
119 ModelBuilder mSubModel;
226 void becomeSingleStep(const std::shared_ptr<Device> device, const ModelBuilder* model);
228 int finish(const ModelBuilder* fromModel, int32_t executionPreference);
264 virtual int finish(const ModelBuilder* fromModel, int32_t executionPreference) = 0;
270 SimpleBody(std::shared_ptr<Device> device, const ModelBuilder* model, in SimpleBody()
275 int finish(const ModelBuilder* fromModel, int32_t executionPreference) override;
[all …]
|
| D | NeuralNetworks.cpp | 638 const ModelBuilder* m = reinterpret_cast<const ModelBuilder*>(model); in ANeuralNetworksModel_getSupportedOperationsForDevices()
717 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksCompilation_createForDevices()
866 ModelBuilder* m = new (std::nothrow) ModelBuilder(); in ANeuralNetworksModel_create()
878 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_free()
888 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_finish()
899 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_addOperand()
910 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValue()
923 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandValueFromMemory()
936 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_addOperation()
949 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model); in ANeuralNetworksModel_setOperandSymmPerChannelQuantParams()
[all …]
|
| D | ModelBuilder.cpp | 38 bool ModelBuilder::badState(const char* name) { in badState()
50 int ModelBuilder::getExtensionType(const char* extensionName, uint16_t typeWithinExtension, in getExtensionType()
57 int ModelBuilder::addOperand(const ANeuralNetworksOperandType& type) { in addOperand()
98 int ModelBuilder::setOperandValue(uint32_t index, const void* buffer, size_t length) { in setOperandValue()
166 int ModelBuilder::setOperandSymmPerChannelQuantParams( in setOperandSymmPerChannelQuantParams()
201 int ModelBuilder::setOperandExtensionData(uint32_t index, const void* data, size_t length) { in setOperandExtensionData()
240 int ModelBuilder::copyLargeValuesToSharedMemory() { in copyLargeValuesToSharedMemory()
278 int ModelBuilder::setOperandValueFromMemory(uint32_t index, const Memory* memory, uint32_t offset, in setOperandValueFromMemory()
319 int ModelBuilder::addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, in addOperation()
364 int ModelBuilder::identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, in identifyInputsAndOutputs()
[all …]
|
| D | CompilationBuilder.h | 32 class ModelBuilder; variable
41 CompilationBuilder(const ModelBuilder* model,
60 const ModelBuilder* mModel;
|
| D | ExecutionBuilder.h | 43 class ModelBuilder; variable
109 const ModelBuilder* getModel() const { return mModel; } in getModel()
129 const ModelBuilder* mModel;
179 StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model,
253 const ModelBuilder* mModel;
|
| D | ExecutionPlan.cpp | 134 int compileModelAndCache(const std::shared_ptr<Device>& device, const ModelBuilder* model, in compileModelAndCache()
159 int compile(std::shared_ptr<Device> device, const ModelBuilder* model, int32_t executionPreference, in compile()
178 int copyOperandExtraParams(ModelBuilder& model, uint32_t toOperandIndex, in copyOperandExtraParams()
214 OperandTracker(const ModelBuilder* model, OperationReadyCallback cb);
221 const ModelBuilder* mModel;
226 OperandTracker::OperandTracker(const ModelBuilder* model, OperationReadyCallback cb) : in OperandTracker()
272 const ModelBuilder& fromModel, OperandKind kind) { in addOperand()
370 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) { in addOperation()
493 int ExecutionStep::finishSubModel(const ModelBuilder* fromModel, bool* hasOutputOfUnknownSize, in finishSubModel()
591 int ExecutionPlan::CompoundBody::finish(const ModelBuilder* fromModel, in finish()
[all …]
|
| D | ModelBuilder.h | 36 class ModelBuilder {
38 ModelBuilder() {} in ModelBuilder() function
|
| D | CompilationBuilder.cpp | 32 CompilationBuilder::CompilationBuilder(const ModelBuilder* model, in CompilationBuilder()
|
| D | Memory.h | 33 class ModelBuilder; variable
|
| D | Android.bp | 46 "ModelBuilder.cpp",
|
| D | ExecutionBuilder.cpp | 716 StepExecutor::StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model, in StepExecutor()
|
| /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 | TestCompliance.cpp | 39 ModelBuilder* modelBuilder = reinterpret_cast<ModelBuilder*>(wrapperModel.getHandle()); in createHidlModel()
|
| D | TestPartitioningRandom.cpp | 101 using ModelBuilder = nn::ModelBuilder; typedef
202 const ModelBuilder* builder() const { in builder()
203 return reinterpret_cast<const ModelBuilder*>(getHandle()); in builder()
223 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in TestCompilation()
443 ModelStats(const ModelBuilder* model) : in ModelStats()
446 mBuilder(reinterpret_cast<const ModelBuilder*>(model->getHandle())) { } in ModelStats()
460 const ModelBuilder* mBuilder;
|
| D | TestPartitioning.cpp | 145 using ModelBuilder = ::android::nn::ModelBuilder; typedef
273 void dump(const char* name, const ModelBuilder* model) { in dump()
579 return reinterpret_cast<ModelBuilder*>(getHandle())->partitionTheWork( in partitionTheWork()
586 const ModelBuilder* mb = reinterpret_cast<const ModelBuilder*>(getHandle()); in dump()
641 ModelBuilder* m = reinterpret_cast<ModelBuilder*>(model->getHandle()); in PartitioningCompilation()
859 void buildDefinitionMap(const ModelBuilder* model, in buildDefinitionMap()
975 bool compare(const ModelBuilder* modelA, const ModelBuilder* modelB, in compare()
1128 reinterpret_cast<const ModelBuilder*>(model->getHandle()), in compare()
|
| D | TestExecution.cpp | 324 nn::ModelBuilder* m = reinterpret_cast<nn::ModelBuilder*>(model->getHandle()); in TestCompilation()
|