Searched refs:tpu_replicated_input_index (Results 1 – 1 of 1) sorted by relevance
418 int64_t tpu_replicated_input_index = tpu_replicated_input.index(); in ReplicateCluster() local421 packed_input_indices.push_back(tpu_replicated_input_index); in ReplicateCluster()426 replicated_input_indices.push_back(tpu_replicated_input_index); in ReplicateCluster()429 if (tpu_replicated_input_index != -1) has_replicated_input_index = true; in ReplicateCluster()