Searched refs:GetAllSlices (Results 1 – 4 of 4) sorted by relevance
44 assignment_.GetAllSlices(&hlo, index); in AddAliasingInformationToIrArray()
382 std::set<BufferAllocation::Slice> GetAllSlices(
357 std::set<BufferAllocation::Slice> BufferAssignment::GetAllSlices( in GetAllSlices() function in xla::BufferAssignment464 auto shape_slices = GetAllSlices(instr, index); in HaveDisjointSlices()1053 assignment->GetAllSlices(operand, /*index=*/{})) { in AssignBuffersForComputation()
1540 auto element_slices = assignment->GetAllSlices(select, /*index=*/{0}); in TEST_F()