Searched refs:tpu_replicated_metadata_node (Results 1 – 1 of 1) sorted by relevance
564 Node* tpu_replicated_metadata_node = nullptr; in FindTpuReplicatedInputAndXlaSharding() local567 tpu_replicated_metadata_node = input_edge->src(); in FindTpuReplicatedInputAndXlaSharding()582 tpu_replicated_metadata_node); in FindTpuReplicatedInputAndXlaSharding()