Searched defs:is_stable (Results 1 – 11 of 11) sorted by relevance
28 int32_t* values_primitive_type_size_in_bytes, bool is_stable, in __xla_cpu_runtime_KeyValueSort()
386 bool is_stable) -> XlaOp { in BuildOpsSubmodule()
247 bool is_stable = 60; field
532 int64_t dimension_to_sort, bool is_stable, HloComputation::Builder* builder, in MakeSortHlo()
908 bool is_stable() const { return is_stable_; } in is_stable() function
1771 optional<bool> is_stable = false; in CreateInstruction() local
1778 bool is_stable) { in CreateSort()
1181 bool is_stable) in HloSortInstruction()
315 bool is_stable) { in SortInternal()
1196 def is_stable(self): member in OnMeshPrefix
2172 bool is_stable) { in Sort()2188 int64_t dimension, bool is_stable) { in SortInternal()4842 int64_t dimension, bool is_stable) { in Sort()