Searched refs:PreparedModel (Results 1 – 7 of 7) 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 | 102 std::pair<int, std::shared_ptr<PreparedModel>> prepareModel(
122 class DriverPreparedModel : public PreparedModel {
244 std::pair<int, std::shared_ptr<PreparedModel>> DriverDevice::prepareModel( in prepareModel()
577 std::pair<int, std::shared_ptr<PreparedModel>> prepareModel(
597 class CpuPreparedModel : public PreparedModel {
602 static std::pair<int, std::shared_ptr<PreparedModel>> create(Model hidlModel);
652 std::pair<int, std::shared_ptr<PreparedModel>> CpuDevice::prepareModel( in prepareModel()
681 std::pair<int, std::shared_ptr<PreparedModel>> CpuPreparedModel::create(Model hidlModel) { in create()
687 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 | 982 std::shared_ptr<PreparedModel> preparedModel, const ExecutionStep* step) in StepExecutor()
|