Searched refs:PreparedModel (Results 1 – 8 of 8) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | Manager.h | 46 class PreparedModel {
47 DISALLOW_COPY_AND_ASSIGN(PreparedModel);
50 PreparedModel() = default;
51 virtual ~PreparedModel() = default;
106 virtual std::pair<int, std::shared_ptr<PreparedModel>> prepareModel(
|
| D | ExecutionBuilder.h | 46 class PreparedModel; variable
230 std::shared_ptr<Device> device, std::shared_ptr<PreparedModel> preparedModel,
304 std::shared_ptr<PreparedModel> mPreparedModel;
|
| D | Memory.h | 46 class PreparedModel; variable
96 using StepRoleCallback = std::function<void(const PreparedModel*, IOType, uint32_t)>;
100 ObjectTracker<PreparedModel> preparedModels;
|
| D | Manager.cpp | 101 std::pair<int, std::shared_ptr<PreparedModel>> prepareModel(
121 class DriverPreparedModel : public PreparedModel {
243 std::pair<int, std::shared_ptr<PreparedModel>> DriverDevice::prepareModel( in prepareModel()
576 std::pair<int, std::shared_ptr<PreparedModel>> prepareModel(
596 class CpuPreparedModel : public PreparedModel {
601 static std::pair<int, std::shared_ptr<PreparedModel>> create(Model hidlModel);
651 std::pair<int, std::shared_ptr<PreparedModel>> CpuDevice::prepareModel( in prepareModel()
680 std::pair<int, std::shared_ptr<PreparedModel>> CpuPreparedModel::create(Model hidlModel) { in create()
686 std::shared_ptr<PreparedModel> preparedModel = in create()
|
| D | ExecutionPlan.h | 54 class PreparedModel; variable
154 std::shared_ptr<PreparedModel> getPreparedStepModel() const { return mPreparedStepModel; } in getPreparedStepModel()
186 std::shared_ptr<PreparedModel> mPreparedStepModel;
659 std::shared_ptr<PreparedModel> mPreparedModel;
|
| D | Memory.cpp | 336 std::vector<std::tuple<const PreparedModel*, IOType, uint32_t>> roles; in addRole()
|
| D | ExecutionBuilder.cpp | 980 std::shared_ptr<PreparedModel> preparedModel, const ExecutionStep* step) in StepExecutor()
|
| D | ExecutionPlan.cpp | 74 std::shared_ptr<PreparedModel>* preparedModel) { in compile()
|