Home
last modified time | relevance | path

Searched refs:UniqueTiledDim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h131 absl::optional<int64> UniqueTiledDim(const HloSharding& sharding);
Dspmd_partitioner.cc2010 auto maybe_input_sharded_dim = UniqueTiledDim(operand.sharding()); in HandleReshape()
2011 auto maybe_output_sharded_dim = UniqueTiledDim(sharding); in HandleReshape()
Dspmd_partitioner_util.cc629 absl::optional<int64> UniqueTiledDim(const HloSharding& sharding) { in UniqueTiledDim() function