Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbuffer_assignment_test.cc268 const BufferAllocation& GetAssignedInputAllocation( in GetAssignedInputAllocation() function in xla::__anond1a23f2e0111::BufferAssignmentTest
474 GetAssignedInputAllocation(*buffers, paramscalar); in TEST_F()
475 BufferAllocation param0_buffer = GetAssignedInputAllocation(*buffers, param0); in TEST_F()
476 BufferAllocation param1_buffer = GetAssignedInputAllocation(*buffers, param1); in TEST_F()
517 BufferAllocation param_buffer = GetAssignedInputAllocation(*buffers, param); in TEST_F()
556 GetAssignedInputAllocation(*buffers, paramscalar); in TEST_F()
557 BufferAllocation param0_buffer = GetAssignedInputAllocation(*buffers, param0); in TEST_F()
558 BufferAllocation param1_buffer = GetAssignedInputAllocation(*buffers, param1); in TEST_F()
618 GetAssignedInputAllocation(*buffers, paramscalar); in TEST_F()
619 BufferAllocation param0_buffer = GetAssignedInputAllocation(*buffers, param0); in TEST_F()
[all …]