Searched refs:kernel_graph_ (Results 1 – 6 of 6) sorted by relevance
394 …CallInfoFinder(const KernelGraphPtr &kg, AscendAutoMonadContext *context) : kernel_graph_(kg), con… in CallInfoFinder()412 kernel_graph_->UpdateChildGraphOrder(); in FindCallSites()415 auto nodes = TopoSort(kernel_graph_->output()); in FindCallSites()433 if (kernel_graph_ != context_.TopGraph() && !call_info->call_sites.empty()) { in FindCallSites()434 auto real_output = GetRealNode(kernel_graph_->output()); in FindCallSites()620 auto [iter, ok] = context_.call_info_map.add(kernel_graph_); in CreateCallInfo()686 const KernelGraphPtr &kernel_graph_; member in mindspore::session::__anon8f02f8710111::CallInfoFinder707 : kernel_graph_(kg), in AscendAutoMonadConverter()731 kernel_graph_->SetExecOrderByDefault(); in Run()733 const auto &nodes = kernel_graph_->execution_order(); in Run()[all …]
28 explicit AscendAutoMonad(NotNull<KernelGraphPtr> kg) : kernel_graph_(kg) {} in AscendAutoMonad()38 NotNull<KernelGraphPtr> kernel_graph_;
66 kernel_graph_ = nullptr; in ~DataDumper()75 MS_EXCEPTION_IF_NULL(kernel_graph_); in GetNeedDumpKernelList()76 for (const auto &kernel : kernel_graph_->execution_order()) { in GetNeedDumpKernelList()106 MS_EXCEPTION_IF_NULL(kernel_graph_); in LoadDumpInfo()111 auto kernels = kernel_graph_->execution_order(); in LoadDumpInfo()129 graph_id_ = kernel_graph_->graph_id(); in LoadDumpInfo()137 MS_EXCEPTION_IF_NULL(kernel_graph_); in SetOpMappingInfo()139 const auto &input_ctrl_tensors = kernel_graph_->input_ctrl_tensors(); in SetOpMappingInfo()148 uint32_t graph_id = kernel_graph_->graph_id(); in SetOpMappingInfo()
51 kernel_graph_(kernel_graph) {} in DataDumper()89 const session::KernelGraph *kernel_graph_; variable
29 kernel_graph_ = kernel_graph; in Init()182 MS_EXCEPTION_IF_NULL(kernel_graph_); in IsCommunicationRelevantOp()183 const auto &graph_manager = kernel_graph_->manager(); in IsCommunicationRelevantOp()203 MS_EXCEPTION_IF_NULL(kernel_graph_); in IsInplaceRelevantOp()204 const auto &graph_manager = kernel_graph_->manager(); in IsInplaceRelevantOp()225 MS_EXCEPTION_IF_NULL(kernel_graph_); in SaveUserKernelTopoOrder()226 const auto &graph_manager = kernel_graph_->manager(); in SaveUserKernelTopoOrder()
163 const mindspore::session::KernelGraph *kernel_graph_{nullptr};