Searched refs:buffer_liveness (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_liveness.h | 38 using Colorer = std::function<Status(const BufferLiveness& buffer_liveness)>;
73 return [](const BufferLiveness& buffer_liveness) { in DefaultColorer()
75 id < buffer_liveness.points_to_analysis().num_logical_buffers(); in DefaultColorer()
77 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in DefaultColorer()
|
| D | buffer_assignment.cc | 1419 const BufferLiveness& buffer_liveness, in MergeColocatedBufferSets() argument
1453 if (buffer_liveness.MaybeLiveOut(*buffer) || in MergeColocatedBufferSets()
1464 auto cannot_merge_buffer_sets = [&colocated_buffer_sets, &buffer_liveness, in MergeColocatedBufferSets()
1484 buffer_liveness.MayInterfere(*buffer_a, *buffer_b)) { in MergeColocatedBufferSets()
1531 const HloModule* module, const BufferLiveness& buffer_liveness, in BuildColocatedBufferSets() argument
1535 buffer_liveness.points_to_analysis(); in BuildColocatedBufferSets()
1686 MergeColocatedBufferSets(*colocated_buffer_sets, buffer_liveness, in BuildColocatedBufferSets()
|
| D | buffer_assignment.h | 618 const HloModule* module, const BufferLiveness& buffer_liveness,
641 const BufferLiveness& buffer_liveness,
|
| D | buffer_assignment_test.cc | 579 auto colorer = [](const BufferLiveness& buffer_liveness) { in TEST_F() argument
583 id < buffer_liveness.points_to_analysis().num_logical_buffers(); in TEST_F()
585 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in TEST_F()
640 auto colorer = [](const BufferLiveness& buffer_liveness) { in TEST_F() argument
642 id < buffer_liveness.points_to_analysis().num_logical_buffers(); in TEST_F()
644 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in TEST_F()
646 buffer_liveness.points_to_analysis().GetBufferAliases(buffer); in TEST_F()
|
| D | BUILD | 1070 name = "buffer_liveness",
1072 "buffer_liveness.cc",
1075 "buffer_liveness.h",
1098 ":buffer_liveness",
1121 ":buffer_liveness",
2631 ":buffer_liveness",
2750 ":buffer_liveness",
3450 ":buffer_liveness",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 113 "//tensorflow/compiler/xla/service:buffer_liveness",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 840 "//tensorflow/compiler/xla/service:buffer_liveness",
|