Searched defs:cast_node (Results 1 – 9 of 9) sorted by relevance
52 auto cast_node = input_node->cast<CNodePtr>(); in Process() local59 void ConvertCastFormat::SetCastFormat(const CNodePtr &cast_node, const string &format) const { in SetCastFormat() argument69 void ConvertCastFormat::ChangeCastFormat(const CNodePtr &cast_node, const FuncGraphPtr &func_graph)… in ChangeCastFormat() argument
27 void UpdateBuildInfo(const AnfNodePtr &matmul_node, const AnfNodePtr &cast_node) { in UpdateBuildInfo()66 auto cast_node = cnode->inputs().back(); // bias node in DoFuse() local
275 auto cast_node = node_input->cast<CNodePtr>(); in ReorderCastUpTypeInsensitive() local292 …if (!std::all_of(cast_nodes.begin(), cast_nodes.end(), [&small_type](const AnfNodePtr &cast_node) { in ReorderCastUpTypeInsensitive()
32 for (auto &cast_node : node_list) { in GetDealList() local
78 auto cast_node = node->cast<CNodePtr>()->input(kInputIndex); in GetCastInput() local
195 auto cast_node = NewCNode({NewValueNode(prim_cast_class)}, graph); in DoCast() local
410 auto cast_node = input_node->cast<CNodePtr>(); in RefreshCastAndParamWeightFormat() local
177 auto cast_node = inputs[0]; in TEST_F() local
1539 CNodePtr cast_node = InsertAllGatherAfterCast(cnode); in InsertAllGatherOp() local