Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1843 int op_parallel_tiles_num = new_operand_shard.NumTiles(operand_parallel_dims); in GatherOperandsShardedAcrossParallelDims() local
1844 if (idx_parallel_tiles_num == 1 && op_parallel_tiles_num == 1) { in GatherOperandsShardedAcrossParallelDims()
1870 if (idx_parallel_tiles_num != op_parallel_tiles_num) { in GatherOperandsShardedAcrossParallelDims()
1875 if (idx_parallel_tiles_num < op_parallel_tiles_num) { in GatherOperandsShardedAcrossParallelDims()