Searched refs:key_fn (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | comparison_util.h | 250 auto LessThanByKey(KeyFn&& key_fn) { in LessThanByKey() argument
251 return [=](const auto& a, const auto& b) { return key_fn(a) < key_fn(b); }; in LessThanByKey()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_combiner_utils.h | 47 const std::function<std::optional<K>(const HloInstruction*)>& key_fn, in CombineInstructionsByKey() argument
56 std::optional<K> key = key_fn(instruction); in CombineInstructionsByKey()
|
| D | all_reduce_combiner.cc | 136 auto key_fn = in Run() local
148 computation, key_fn, &CombineAllReduces, in Run()
|
| D | reduce_scatter_combiner.cc | 141 auto key_fn = [&domain_map](const HloInstruction* instruction) in Run() local
156 computation, key_fn, &CombineReduceScatters, in Run()
|
| D | all_gather_combiner.cc | 163 auto key_fn = [&domain_map](const HloInstruction* instruction) { in Run() local
170 computation, key_fn, &CombineAllGathers, in Run()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants_test.py | 87 def merge_lists(repeated1, repeated2, empty_fn, key_fn, merge_fn): argument
90 xs1 = {key_fn(x): x for x in repeated1}
91 xs2 = {key_fn(x): x for x in repeated2}
96 return sorted(merged.values(), key=key_fn)
|