Home
last modified time | relevance | path

Searched refs:ScatterUpdateShardingFromOutput (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h168 absl::optional<HloSharding> ScatterUpdateShardingFromOutput(
Dhlo_sharding_util.cc1031 absl::optional<HloSharding> ScatterUpdateShardingFromOutput( in ScatterUpdateShardingFromOutput() function
Dsharding_propagation.cc1385 return hlo_sharding_util::ScatterUpdateShardingFromOutput( in GetShardingFromUser()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc617 auto maybe_passthrough = hlo_sharding_util::ScatterUpdateShardingFromOutput( in HandleScatter()