Searched refs:HasTopLevelAllocation (Results 1 – 5 of 5) sorted by relevance
281 if (!buffers.HasTopLevelAllocation(hlo)) { in ValidateBuffers()316 if (assignment.HasTopLevelAllocation(instruction)) { in BuffersDistinct()323 if (assignment.HasTopLevelAllocation(instruction)) { in BuffersDistinct()343 EXPECT_TRUE(buffers->HasTopLevelAllocation(const0)); in TEST_F()348 EXPECT_FALSE(buffers->HasTopLevelAllocation(const0)); in TEST_F()367 EXPECT_TRUE(buffers->HasTopLevelAllocation(const0)); in TEST_F()368 EXPECT_TRUE(buffers->HasTopLevelAllocation(const1)); in TEST_F()373 EXPECT_FALSE(buffers->HasTopLevelAllocation(const0)); in TEST_F()374 EXPECT_FALSE(buffers->HasTopLevelAllocation(const1)); in TEST_F()397 EXPECT_EQ(buffers->HasTopLevelAllocation(tuple), in TEST_F()[all …]
392 bool HasTopLevelAllocation(const HloInstruction* instruction) const;
405 bool BufferAssignment::HasTopLevelAllocation( in HasTopLevelAllocation() function in xla::BufferAssignment
30 return assignment.HasTopLevelAllocation(dynamic_update_slice) && in CanUpdateDynamicSliceInPlace()31 assignment.HasTopLevelAllocation(operand) && in CanUpdateDynamicSliceInPlace()
92 if (!buffer_assignment_->HasTopLevelAllocation(non_io_hlo)) { in EmitBasePointersForHlos()