Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h274 absl::optional<int64> GetKValueInTopKWhenPartitionSortDim(HloInstruction* hlo);
Dspmd_partitioner_util.cc1155 absl::optional<int64> GetKValueInTopKWhenPartitionSortDim(HloInstruction* hlo) { in GetKValueInTopKWhenPartitionSortDim() function
Dspmd_partitioner.cc1645 auto k = GetKValueInTopKWhenPartitionSortDim(hlo); in HandleSort()