Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment_test.cc281 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 …]
Dbuffer_assignment.h392 bool HasTopLevelAllocation(const HloInstruction* instruction) const;
Dbuffer_assignment.cc405 bool BufferAssignment::HasTopLevelAllocation( in HasTopLevelAllocation() function in xla::BufferAssignment
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc30 return assignment.HasTopLevelAllocation(dynamic_update_slice) && in CanUpdateDynamicSliceInPlace()
31 assignment.HasTopLevelAllocation(operand) && in CanUpdateDynamicSliceInPlace()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc92 if (!buffer_assignment_->HasTopLevelAllocation(non_io_hlo)) { in EmitBasePointersForHlos()