Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment_test.cc261 const BufferAllocation& GetAllocation(const BufferAssignment& buffers, in GetAllocation() function in xla::__anon89e05a210111::BufferAssignmentTest
495 BufferAllocation neg_1_buffer = GetAllocation(*buffers, neg_1, {}); in TEST_F()
496 BufferAllocation neg_2_buffer = GetAllocation(*buffers, neg_2, {}); in TEST_F()
904 auto while_op_allocation = GetAllocation(*buffers, while_op, index); in TEST_F()
905 auto body_root_allocation = GetAllocation(*buffers, body_root, index); in TEST_F()
1290 auto allocation = GetAllocation(*assignment, tuple_param, index); in TEST_F()
1318 GetAllocation(*assignment, tuple_param, /*index=*/{}).maybe_live_out()); in TEST_F()
1322 GetAllocation(*assignment, tuple_param, /*index=*/{1}).maybe_live_out()); in TEST_F()
1323 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 0}) in TEST_F()
1325 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 1}) in TEST_F()
[all …]
Dbuffer_assignment.h373 const BufferAllocation& GetAllocation(BufferAllocation::Index index) const;
Dbuffer_assignment.cc349 return GetAllocation(allocation_index_for_buffer_.at(&buffer)); in GetAssignedAllocation()
368 const BufferAllocation& BufferAssignment::GetAllocation( in GetAllocation() function in xla::BufferAssignment
391 return const_cast<BufferAllocation*>(&GetAllocation(index)); in GetMutableAllocation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.cc48 const BufferAllocation& allocation = buffer_assignment->GetAllocation(i); in Build()
135 const BufferAllocation& allocation = buffer_assignment_->GetAllocation(i); in TearDown()
Dgpu_executable.cc207 const BufferAllocation& allocation = assignment_->GetAllocation(i); in ResolveConstantGlobals()
253 const BufferAllocation& allocation = assignment_->GetAllocation(i); in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc88 auto& allocation = assignment_->GetAllocation(i); in CreateBufferTable()