Searched refs:GetAssignedInputAllocation (Results 1 – 1 of 1) sorted by relevance
249 const BufferAllocation& GetAssignedInputAllocation( in GetAssignedInputAllocation() function in xla::__anon3198ac380111::BufferAssignmentTest457 GetAssignedInputAllocation(*buffers, paramscalar); in TEST_F()458 BufferAllocation param0_buffer = GetAssignedInputAllocation(*buffers, param0); in TEST_F()459 BufferAllocation param1_buffer = GetAssignedInputAllocation(*buffers, param1); in TEST_F()500 BufferAllocation param_buffer = GetAssignedInputAllocation(*buffers, param); in TEST_F()539 GetAssignedInputAllocation(*buffers, paramscalar); in TEST_F()540 BufferAllocation param0_buffer = GetAssignedInputAllocation(*buffers, param0); in TEST_F()541 BufferAllocation param1_buffer = GetAssignedInputAllocation(*buffers, param1); in TEST_F()599 GetAssignedInputAllocation(*buffers, paramscalar); in TEST_F()600 BufferAllocation param0_buffer = GetAssignedInputAllocation(*buffers, param0); in TEST_F()[all …]