Searched refs:TimeBound (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_live_range_test.cc | 36 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 …]
|
D | hlo_live_range.h | 52 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_;
|
D | hlo_live_range.cc | 44 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()
|
D | memory_space_assignment.cc | 1325 const HloLiveRange::TimeBound& computation_span = in AllocateAllocationValues() 3331 const HloLiveRange::TimeBound& time_bound = in VerifyAndExportHeapSimulatorTrace()
|
D | memory_space_assignment_test.cc | 3862 HloLiveRange::TimeBound time_bound = in TEST_P()
|