Home
last modified time | relevance | path

Searched refs:HasTopLevelAllocation (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc61 return assignment.HasTopLevelAllocation(dynamic_update_slice) && in CanUpdateDynamicSliceInPlace()
62 assignment.HasTopLevelAllocation(operand) && in CanUpdateDynamicSliceInPlace()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment_test.cc288 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 …]
Dbuffer_assignment.h412 bool HasTopLevelAllocation(const HloInstruction* instruction) const;
Dbuffer_assignment.cc467 bool BufferAssignment::HasTopLevelAllocation( in HasTopLevelAllocation() function in xla::BufferAssignment