Searched refs:tbe_nodes (Results 1 – 3 of 3) sorted by relevance
75 std::vector<AnfNodePtr> tbe_nodes; in KernelBuildParallelCompile() local87 tbe_nodes.push_back(anf_node); in KernelBuildParallelCompile()105 if (!tbe_nodes.empty()) { in KernelBuildParallelCompile()108 tbe_ret = kernel::TbeOpParallelBuild(tbe_nodes); in KernelBuildParallelCompile()111 tbe_ret = build_manager.AscendSingleOpCompile(tbe_nodes); in KernelBuildParallelCompile()
69 …des(const std::shared_ptr<session::KernelGraph> &kernel_graph, std::vector<AnfNodePtr> *tbe_nodes);
518 std::vector<AnfNodePtr> *tbe_nodes) { in GetAllAscendNodes() argument529 tbe_nodes->push_back(anf_node); in GetAllAscendNodes()