Searched refs:GetAllSlices (Results 1 – 3 of 3) sorted by relevance
402 std::set<BufferAllocation::Slice> GetAllSlices(
418 std::set<BufferAllocation::Slice> BufferAssignment::GetAllSlices( in GetAllSlices() function in xla::BufferAssignment526 auto shape_slices = GetAllSlices(instr, index); in HaveDisjointSlices()1170 assignment->GetAllSlices(operand, /*index=*/{})) { in AssignSingleHloBuffer()
1712 auto element_slices = assignment->GetAllSlices(select, /*index=*/{0}); in TEST_F()