Searched refs:backend_ (Results 1 – 9 of 9) sorted by relevance
43 explicit CommManager(string backend) : backend_(std::move(backend)) {} in CommManager()44 string backend_; variable
82 std::string backend() const { return backend_; } in backend()97 std::string backend_; variable
189 backend_ = backend; in Init()193 } else if (backend_ == NCCL_BACKEND) { in Init()
99 BackendPtr backend_; variable132 BackendPtr backend_; variable
69 …nst InstSet &insts, const BackendPtr &backend) : insts_(insts), pc_(0), sp_(0), backend_(backend) { in FinalVM()315 MS_EXCEPTION_IF_NULL(backend_); in InstRealSwitch()316 if (backend_->GetCond(c, &bool_value)) { in InstRealSwitch()349 MS_EXCEPTION_IF_NULL(backend_); in InstSwitchLayer()350 if (!backend_->GetIndex(index, &idx_value)) { in InstSwitchLayer()
60 …leGraph(const BackendPtr &backend, const std::vector<PrimitivePtr> &cut_list) : backend_(backend) { in CompileGraph()61 MS_EXCEPTION_IF_NULL(backend_); in CompileGraph()62 lin_convert_ = backend_->convert_fn(); in CompileGraph()498 …eGraphs(const BackendPtr &backend, const std::vector<PrimitivePtr> &cut_list) : backend_(backend) { in CompileGraphs()534 FinalVMPtr rt = std::make_shared<FinalVM>(insts_, backend_); in Link()
143 BackendPtr backend_;
48 std::shared_ptr<compile::MsBackend> backend_; variable
110 backend_ = std::move(backend); in Build()