Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dbuffer_assignment_util.cc25 for (const auto& buffer_offset_pair : allocation.assigned_buffers()) { in InstrForConstantBufferAllocation()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h156 assigned_buffers() const { in assigned_buffers() function
Dbuffer_assignment_test.cc1533 EXPECT_EQ(1, select_alloc.assigned_buffers().size()); in TEST_F()
1535 select_alloc.assigned_buffers().begin()->first->instruction()); in TEST_F()
1721 ASSERT_EQ(buffer.assigned_buffers().size(), 4); in TEST_F()
1834 allocation_for_const_1.assigned_buffers()) { in TEST_F()
1847 allocation_for_const_2.assigned_buffers()) { in TEST_F()
Dbuffer_assignment.cc516 CHECK(allocation->is_reusable() || allocation->assigned_buffers().empty()) in AddAssignment()
862 for (const auto& buffer_offset_size : allocation->assigned_buffers()) { in MaybeAssignBuffer()
1250 for (const auto& pair : allocation.assigned_buffers()) { in ComputePeakMemoryLogicalBuffers()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2874 const auto& assigned_buffers = allocation.assigned_buffers(); in EmitThreadLocalBufferPointer() local
2875 CHECK_EQ(1, assigned_buffers.size()); in EmitThreadLocalBufferPointer()
2876 const Shape& shape = assigned_buffers.begin()->first->shape(); in EmitThreadLocalBufferPointer()