Home
last modified time | relevance | path

Searched refs:maybe_live_out (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.cc47 if ((allocation.maybe_live_out() && in TearDown()
Dgpu_executable.cc340 CHECK(allocation.maybe_live_out() || allocation.IsPreallocatedTempBuffer()); in BufferForAllocation()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h141 bool maybe_live_out() const { return maybe_live_out_; } in maybe_live_out() function
221 return is_entry_computation_parameter() || maybe_live_out(); in IsInputOrOutput()
231 !maybe_live_out() && in IsPreallocatedTempBuffer()
Dbuffer_assignment_test.cc1407 EXPECT_FALSE(map_param_alloc.maybe_live_out()); in TEST_F()
1412 EXPECT_FALSE(map_root_alloc.maybe_live_out()); in TEST_F()
1418 EXPECT_FALSE(call_param_alloc.maybe_live_out()); in TEST_F()
1429 EXPECT_FALSE(param_alloc.maybe_live_out()); in TEST_F()
1434 EXPECT_TRUE(map_alloc.maybe_live_out()); in TEST_F()
1465 EXPECT_TRUE(allocation.maybe_live_out()); in TEST_F()
1490 GetAllocation(*assignment, tuple_param, /*index=*/{}).maybe_live_out()); in TEST_F()
1494 GetAllocation(*assignment, tuple_param, /*index=*/{1}).maybe_live_out()); in TEST_F()
1496 .maybe_live_out()); in TEST_F()
1498 .maybe_live_out()); in TEST_F()
[all …]
Dbuffer_assignment.cc360 if (maybe_live_out()) { in ToString()
711 if (allocation.maybe_live_out()) { in ComputeSummaryStats()
1013 if (allocation->maybe_live_out()) { in MaybeAssignBuffer()
Dhlo.proto444 bool maybe_live_out = 7; field