Home
last modified time | relevance | path

Searched refs:TimeBound (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range_test.cc36 using TimeBound = HloLiveRange::TimeBound; typedef
63 HloLiveRange::TimeBound LiveRangeAt(const HloInstruction* instruction, in LiveRangeAt()
106 EXPECT_EQ(LiveRangeAt(paramA), TimeBound({0, 3})); in TEST_F()
107 EXPECT_EQ(LiveRangeAt(paramX), TimeBound({0, 3})); in TEST_F()
109 EXPECT_EQ(LiveRangeAt(mul), TimeBound({2, 3})); in TEST_F()
136 EXPECT_EQ(LiveRangeAt(paramA), TimeBound({0, 5})); in TEST_F()
137 EXPECT_EQ(LiveRangeAt(paramX), TimeBound({0, 5})); in TEST_F()
138 EXPECT_EQ(LiveRangeAt(paramY), TimeBound({0, 5})); in TEST_F()
141 EXPECT_EQ(LiveRangeAt(mul), TimeBound({2, 4})); in TEST_F()
143 EXPECT_EQ(LiveRangeAt(add), TimeBound({4, 5})); in TEST_F()
[all …]
Dhlo_live_range.h52 struct TimeBound { struct
56 bool friend operator==(const TimeBound& a, const TimeBound& b) { argument
59 bool friend operator!=(const TimeBound& a, const TimeBound& b) {
77 const absl::flat_hash_map<const HloValue*, TimeBound>& buffer_live_ranges() in buffer_live_ranges()
82 absl::flat_hash_map<const HloValue*, TimeBound>& buffer_live_ranges() { in buffer_live_ranges()
87 const absl::flat_hash_map<const HloComputation*, TimeBound>&
205 absl::flat_hash_map<const HloComputation*, TimeBound> computation_span_times_;
206 absl::flat_hash_map<const HloValue*, TimeBound> buffer_live_ranges_;
Dhlo_live_range.cc44 const TimeBound& live_range1 = buffer_live_ranges_.at(value1); in NormalizeAliasedBuffers()
45 const TimeBound& live_range2 = buffer_live_ranges_.at(value2); in NormalizeAliasedBuffers()
54 TimeBound& live_range1 = buffer_live_ranges_[value1]; in NormalizeAliasedBuffers()
55 TimeBound& live_range2 = buffer_live_ranges_[value2]; in NormalizeAliasedBuffers()
123 TimeBound{start_time, time}); in FlattenSchedule()
Dmemory_space_assignment.cc1325 const HloLiveRange::TimeBound& computation_span = in AllocateAllocationValues()
3331 const HloLiveRange::TimeBound& time_bound = in VerifyAndExportHeapSimulatorTrace()
Dmemory_space_assignment_test.cc3862 HloLiveRange::TimeBound time_bound = in TEST_P()