Searched refs:mul_graph_ (Results 1 – 2 of 2) sorted by relevance
41 if (mul_graph_ == nullptr) { in ObtainLaunchMulKernelMod()43 mul_graph_ = ObtainMulKernelGraph(); in ObtainLaunchMulKernelMod()44 MS_EXCEPTION_IF_NULL(mul_graph_); in ObtainLaunchMulKernelMod()46 KernelSelect(mul_graph_); in ObtainLaunchMulKernelMod()48 KernelBuild(mul_graph_); in ObtainLaunchMulKernelMod()51 if (mul_graph_->execution_order().size() != 1) { in ObtainLaunchMulKernelMod()53 << mul_graph_->execution_order().size() << " nodes."; in ObtainLaunchMulKernelMod()55 return AnfAlgo::GetKernelMod(mul_graph_->execution_order()[0]); in ObtainLaunchMulKernelMod()
33 mul_graph_(nullptr) {} in LaunchMul()54 std::shared_ptr<session::KernelGraph> mul_graph_; variable