Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h184 std::vector<int64> DevicesForSharding(
Dhlo_sharding_util.cc1125 std::vector<int64> DevicesForSharding( in DevicesForSharding() function
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3766 TF_RET_CHECK(num_partitions_ == hlo_sharding_util::DevicesForSharding( in PreprocessSharding()