Searched refs:GetTopLevelAllocation (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 256 const BufferAllocation& buffer = GetTopLevelAllocation(buffers, hlo); in GetAssignedOutputAllocation()
266 const BufferAllocation& GetTopLevelAllocation(const BufferAssignment& buffers, in GetTopLevelAllocation() function in xla::__anon3198ac380111::BufferAssignmentTest
458 const BufferAllocation& mul_buffer = GetTopLevelAllocation(*buffers, mul); in TEST_F()
462 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
541 const BufferAllocation& mul_buffer = GetTopLevelAllocation(*buffers, mul); in TEST_F()
546 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
603 const BufferAllocation& mul_buffer = GetTopLevelAllocation(*buffers, mul); in TEST_F()
607 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
672 const BufferAllocation& mul_buffer = GetTopLevelAllocation(*buffers, mul); in TEST_F()
676 const BufferAllocation& add_buffer = GetTopLevelAllocation(*buffers, add); in TEST_F()
[all …]
|