HomeSort by relevance Sort by last modified time
    Searched refs:HasTopLevelAllocation (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
dynamic_update_slice_util.cc 30 return assignment.HasTopLevelAllocation(dynamic_update_slice) &&
31 assignment.HasTopLevelAllocation(operand) &&
  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment_test.cc 281 if (!buffers.HasTopLevelAllocation(hlo)) {
316 if (assignment.HasTopLevelAllocation(instruction)) {
323 if (assignment.HasTopLevelAllocation(instruction)) {
343 EXPECT_TRUE(buffers->HasTopLevelAllocation(const0));
348 EXPECT_FALSE(buffers->HasTopLevelAllocation(const0));
367 EXPECT_TRUE(buffers->HasTopLevelAllocation(const0));
368 EXPECT_TRUE(buffers->HasTopLevelAllocation(const1));
373 EXPECT_FALSE(buffers->HasTopLevelAllocation(const0));
374 EXPECT_FALSE(buffers->HasTopLevelAllocation(const1));
395 // Make sure that HasAllocationAt() agrees with what HasTopLevelAllocation()
    [all...]
buffer_assignment.h 392 bool HasTopLevelAllocation(const HloInstruction* instruction) const;
425 // REQUIRES: HasTopLevelAllocation(hlo_a) && HasTopLevelAllocation(hlo_b).
buffer_assignment.cc 405 bool BufferAssignment::HasTopLevelAllocation(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
hlo_to_ir_bindings.cc 92 if (!buffer_assignment_->HasTopLevelAllocation(non_io_hlo)) {

Completed in 212 milliseconds