Searched refs:GetTopLevelAllocation (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 263 const BufferAllocation& buffer = GetTopLevelAllocation(buffers, hlo); in GetAssignedOutputAllocation()
273 const BufferAllocation& GetTopLevelAllocation(const BufferAssignment& buffers, in GetTopLevelAllocation() function in xla::__anon3198ac380111::BufferAssignmentTest
465 const BufferAllocation& mul_buffer = GetTopLevelAllocation(*buffers, mul); in TEST_F()
469 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
547 const BufferAllocation& sub_buffer = GetTopLevelAllocation(*buffers, sub); in TEST_F()
552 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
607 const BufferAllocation& mul_buffer = GetTopLevelAllocation(*buffers, mul); in TEST_F()
611 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
681 const BufferAllocation& mul_buffer = GetTopLevelAllocation(*buffers, mul); in TEST_F()
685 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
[all …]
|