Searched defs:tpu_replicated_input_ops (Results 1 – 1 of 1) sorted by relevance
555 TpuReplicatedInputInfoMap& tpu_replicated_input_ops) { in FindTpuReplicatedInputAndXlaSharding()2205 TpuReplicatedInputInfoMap tpu_replicated_input_ops; in OptimizeTpuInputOutputTensors() local