Searched defs:xla_sharding_ops (Results 1 – 1 of 1) sorted by relevance
538 const Graph* graph, XlaShardingInfoMap& xla_sharding_ops, in FindTpuReplicatedInputAndXlaSharding()2113 XlaShardingInfoMap xla_sharding_ops; in OptimizeTpuInputOutputTensors() local