Searched refs:assigned_buffers (Results 1 – 5 of 5) sorted by relevance
25 for (const auto& buffer_offset_pair : allocation.assigned_buffers()) { in InstrForConstantBufferAllocation()
308 for (const auto& p : alloc.assigned_buffers()) { in GetEntryParameterInstruction()323 for (const auto& p : alloc.assigned_buffers()) { in GetOutputInstruction()575 CHECK(allocation->is_reusable() || allocation->assigned_buffers().empty()) in AddAssignment()785 for (const auto& p : allocation.assigned_buffers()) { in ToString()823 absl::c_copy(allocation.assigned_buffers(), std::back_inserter(buffers)); in BufferInfoString()1030 for (const auto& buffer_offset_size : allocation->assigned_buffers()) { in MaybeAssignBuffer()1046 for (const auto& buffer_offset_size : allocation->assigned_buffers()) { in MaybeAssignBuffer()1357 absl::flat_hash_set<const HloBuffer*>* assigned_buffers, in AssignPresetBuffers() argument1397 assigned_buffers->insert(&buffer); in AssignPresetBuffers()1510 for (const auto& pair : allocation.assigned_buffers()) { in ComputePeakMemoryLogicalBuffers()
160 const absl::flat_hash_map<const HloValue*, OffsetSize>& assigned_buffers() in assigned_buffers() function658 absl::flat_hash_set<const HloBuffer*>* assigned_buffers,
757 EXPECT_EQ(mul_buffer.assigned_buffers().size(), 2); in TEST_F()758 for (const auto& value_and_offsetsize : mul_buffer.assigned_buffers()) { in TEST_F()847 EXPECT_EQ(data_buffer.assigned_buffers().size(), 5); in TEST_F()848 for (const auto& value_and_offsetsize : data_buffer.assigned_buffers()) { in TEST_F()1705 EXPECT_EQ(1, select_alloc.assigned_buffers().size()); in TEST_F()1707 select_alloc.assigned_buffers().begin()->first->instruction()); in TEST_F()1893 ASSERT_EQ(buffer.assigned_buffers().size(), 4); in TEST_F()1983 allocation_for_const_1.assigned_buffers()) { in TEST_F()1996 allocation_for_const_2.assigned_buffers()) { in TEST_F()
3055 const auto& assigned_buffers = allocation.assigned_buffers(); in EmitThreadLocalBufferPointer() local3056 CHECK_EQ(1, assigned_buffers.size()); in EmitThreadLocalBufferPointer()3057 const Shape& shape = assigned_buffers.begin()->first->shape(); in EmitThreadLocalBufferPointer()