Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc808 int64_t sharding_index = 0; in GetSubSharding() local
812 sharding_index += in GetSubSharding()
818 auto begin_it = tuple_elements_.begin() + sharding_index; in GetSubSharding()
823 return tuple_elements_[sharding_index]; in GetSubSharding()
Dsharding_propagation.cc1466 int64_t sharding_index = 0; in GetShardingFromUser() local
1472 sharding_index += 1; in GetShardingFromUser()
1474 sharding_index += in GetShardingFromUser()
1487 new_sharding.tuple_elements()[sharding_index] = user.sharding(); in GetShardingFromUser()
1501 new_sharding.tuple_elements()[sharding_index + i] = in GetShardingFromUser()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc324 const int sharding_index = output_sharding_and_index.index(); in ParseAndValidateOutputSharding() local
327 "non-string output sharding at index {0}", sharding_index)); in ParseAndValidateOutputSharding()