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.cc628 int64 sharding_index = 0; in GetSubSharding() local
632 sharding_index += in GetSubSharding()
638 auto begin_it = tuple_elements_.begin() + sharding_index; in GetSubSharding()
643 return tuple_elements_[sharding_index]; in GetSubSharding()
Dsharding_propagation.cc1282 int64 sharding_index = 0; in GetShardingFromUser() local
1288 sharding_index += 1; in GetShardingFromUser()
1290 sharding_index += in GetShardingFromUser()
1295 new_sharding.tuple_elements()[sharding_index] = user.sharding(); in GetShardingFromUser()
1298 new_sharding.tuple_elements()[sharding_index + i] = in GetShardingFromUser()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc330 const int sharding_index = output_sharding_and_index.index(); in ParseAndValidateOutputSharding() local
333 "non-string output sharding at index {0}", sharding_index)); in ParseAndValidateOutputSharding()