Searched refs:HasTopLevelAllocation (Results 1 – 4 of 4) sorted by relevance
61 return assignment.HasTopLevelAllocation(dynamic_update_slice) && in CanUpdateDynamicSliceInPlace()62 assignment.HasTopLevelAllocation(operand) && in CanUpdateDynamicSliceInPlace()
288 if (!buffers.HasTopLevelAllocation(hlo)) { in ValidateBuffers()323 if (assignment.HasTopLevelAllocation(instruction)) { in BuffersDistinct()330 if (assignment.HasTopLevelAllocation(instruction)) { in BuffersDistinct()350 EXPECT_TRUE(buffers->HasTopLevelAllocation(const0)); in TEST_F()355 EXPECT_FALSE(buffers->HasTopLevelAllocation(const0)); in TEST_F()374 EXPECT_TRUE(buffers->HasTopLevelAllocation(const0)); in TEST_F()375 EXPECT_TRUE(buffers->HasTopLevelAllocation(const1)); in TEST_F()380 EXPECT_FALSE(buffers->HasTopLevelAllocation(const0)); in TEST_F()381 EXPECT_FALSE(buffers->HasTopLevelAllocation(const1)); in TEST_F()404 EXPECT_EQ(buffers->HasTopLevelAllocation(tuple), in TEST_F()[all …]
412 bool HasTopLevelAllocation(const HloInstruction* instruction) const;
467 bool BufferAssignment::HasTopLevelAllocation( in HasTopLevelAllocation() function in xla::BufferAssignment