Searched refs:tpu_partitioned_input_node (Results 1 – 1 of 1) sorted by relevance
1839 Node* tpu_partitioned_input_node = graph->AddNode(node_def, &s); in ReplaceAndPartitionXLAShardingVariable() local1845 graph->AddEdge(per_core_vars[core_index], 0, tpu_partitioned_input_node, in ReplaceAndPartitionXLAShardingVariable()1858 {tpu_partitioned_input_node->name(), 0, DT_RESOURCE}); in ReplaceAndPartitionXLAShardingVariable()1868 graph->AddEdge(tpu_partitioned_input_node, 0, tpu_replicated_input_node, 0); in ReplaceAndPartitionXLAShardingVariable()