Searched defs:is_stable (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_key_value_sort.cc | 30 tensorflow::int32* values_primitive_type_size_in_bytes, bool is_stable, in __xla_cpu_runtime_KeyValueSort()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 270 bool is_stable) -> XlaOp { in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 192 bool is_stable = 60; field
|
| D | hlo_creation_utils.cc | 418 int64 dimension_to_sort, bool is_stable, HloComputation::Builder* builder, in MakeSortHlo()
|
| D | hlo_instructions.h | 645 bool is_stable() const { return is_stable_; } in is_stable() function
|
| D | hlo_parser.cc | 1397 optional<bool> is_stable = false; in ParseInstructionRhs() local
|
| D | hlo_instruction.cc | 1479 bool is_stable) { in CreateSort()
|
| D | hlo_instructions.cc | 948 bool is_stable) in HloSortInstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 243 int64 dimension, bool is_stable) { in SortInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2061 bool is_stable) { in Sort()
2077 int64 dimension, bool is_stable) { in SortInternal()
4819 int64 dimension, bool is_stable) { in Sort()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 2612 ValueRange operands, int64_t dimension, bool is_stable) { in build()
|