Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.cc76 if (allocation.maybe_live_out() || allocation.IsPreallocatedTempBuffer()) { in Build()
97 if (allocation.IsPreallocatedTempBuffer()) { in Build()
141 allocation.IsPreallocatedTempBuffer()) { in TearDown()
Dir_emitter_unnested.cc231 if (alloc->IsPreallocatedTempBuffer()) { in BuildKernelPrototype()
1617 if (alloc.IsPreallocatedTempBuffer()) { in BuildKernelThunk()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h221 bool IsPreallocatedTempBuffer() const { in IsPreallocatedTempBuffer() function
Dbuffer_assignment.cc303 if (IsPreallocatedTempBuffer()) { in ToString()
542 return !allocation.IsPreallocatedTempBuffer(); in CombineTempAllocations()
623 if (allocation.IsPreallocatedTempBuffer()) { in ComputeSummaryStats()
Dbuffer_assignment_test.cc1720 EXPECT_TRUE(buffer.IsPreallocatedTempBuffer()); in TEST_F()