Searched refs:assign_value_node (Results 1 – 1 of 1) sorted by relevance
1482 Node* assign_value_node = init_const; in InitializeShardedVarOnTPU() local1523 assign_value_node = split_node; in InitializeShardedVarOnTPU()1538 init_graph->AddEdge(assign_value_node, i, init_assign, 1); in InitializeShardedVarOnTPU()1540 init_graph->AddEdge(assign_value_node, 0, init_assign, 1); in InitializeShardedVarOnTPU()