Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.cc49 allocation.IsPreallocatedTempBuffer()) { in TearDown()
Dgpu_executable.cc340 CHECK(allocation.maybe_live_out() || allocation.IsPreallocatedTempBuffer()); in BufferForAllocation()
Dir_emitter_unnested.cc600 if (alloc->IsPreallocatedTempBuffer()) { in BuildKernelPrototype()
3357 if (alloc.IsPreallocatedTempBuffer()) { in BuildKernelThunkFromBufferSlices()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h226 bool IsPreallocatedTempBuffer() const { in IsPreallocatedTempBuffer() function
Dbuffer_assignment.cc363 if (IsPreallocatedTempBuffer()) { in ToString()
621 return !allocation.IsPreallocatedTempBuffer(); in CombineTempAllocations()
715 if (allocation.IsPreallocatedTempBuffer()) { in ComputeSummaryStats()
Dbuffer_assignment_test.cc1892 EXPECT_TRUE(buffer.IsPreallocatedTempBuffer()); in TEST_F()