Searched refs:GetProcessor (Results 1 – 16 of 16) sorted by relevance
105 std::string GetProcessor(const AnfNodePtr &anf_node) { in GetProcessor() function108 switch (AnfAlgo::GetProcessor(anf_node)) { in GetProcessor()
35 std::string GetProcessor(const AnfNodePtr &anf_node);
120 task_info.processor = tbe::GetProcessor(anf_node); in SavePreBuildTaskInfo()136 task_info.processor = tbe::GetProcessor(anf_node); in SaveTaskInfo()224 task_info.processor = tbe::GetProcessor(anf_node); in SaveSameOpInfo()
86 std::string GetProcessor(const AnfNodePtr &anf_node);87 Processor GetProcessor(const string &processor);
514 Processor GetProcessor(const string &processor) { in GetProcessor() function522 std::string GetProcessor(const AnfNodePtr &anf_node) { in GetProcessor() function525 switch (AnfAlgo::GetProcessor(anf_node)) { in GetProcessor()
45 builder.SetProcessor(AnfAlgo::GetProcessor(cnode)); in GenerateKernelBuildInfo()
77 builder.SetProcessor(AnfAlgo::GetProcessor(cnode)); in GenerateKernelBuildInfo()
54 info_builder.SetProcessor(AnfAlgo::GetProcessor(node)); in CreateReduceSum()
129 info_builder.SetProcessor(AnfAlgo::GetProcessor(orig_node)); in SetNodeInfo()
126 …if (AnfAlgo::GetProcessor(node) != kernel::Processor::AICORE && !IsPrimitiveCNode(node, prim::kPri… in IsClusterableOp()
296 graph_info_builder.SetProcessor(AnfAlgo::GetProcessor(node)); in BuildSelectKernelBuildInfo()
140 if (AnfAlgo::IsRealKernel(item) && (AnfAlgo::GetProcessor(item) != 0)) { in OutputCheck()
653 TEST_F(AnfRuntimeAlgorithmTest, GetProcessor) { in TEST_F() argument665 EXPECT_EQ(AnfAlgo::GetProcessor(add), kernel::AICORE); in TEST_F()666 EXPECT_THROW(AnfAlgo::GetProcessor(nullptr), std::runtime_error); in TEST_F()
221 static kernel::Processor GetProcessor(const AnfNodePtr &node);
1350 kernel::Processor AnfRuntimeAlgorithm::GetProcessor(const AnfNodePtr &node) { in GetProcessor() function in mindspore::session::AnfRuntimeAlgorithm
360 Processor p = kernel::GetProcessor(processor); in DecodeCNode()