Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter.cc27 static bool IsNanSafeGt(HloComputation* comp) { in IsNanSafeGt() function
88 if (!IsNanSafeGt(sort->to_apply())) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h271 bool IsNanSafeGt(HloComputation* computation);
Dspmd_partitioner_util.cc1109 bool IsNanSafeGt(HloComputation* comp) { in IsNanSafeGt() function
1160 if (!IsNanSafeGt(sort->to_apply())) { in GetKValueInTopKWhenPartitionSortDim()