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

  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
dynamic_update_slice_util.h 61 return assignment.HasAllocationAt(operand, index) &&
62 assignment.HasAllocationAt(fusion, {}) &&
  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment.h 387 bool HasAllocationAt(const HloInstruction* instruction,
buffer_assignment_test.cc 379 TEST_F(BufferAssignmentTest, HasAllocationAt) {
381 // HasAllocationAt works correctly.
395 // Make sure that HasAllocationAt() agrees with what HasTopLevelAllocation()
398 buffers->HasAllocationAt(tuple, /*index=*/{}));
400 buffers->HasAllocationAt(tuple, /*index=*/{0}));
402 buffers->HasAllocationAt(tuple, /*index=*/{1}));
404 buffers->HasAllocationAt(tuple, /*index=*/{2}));
    [all...]
buffer_assignment.cc 394 bool BufferAssignment::HasAllocationAt(const HloInstruction* instruction,
407 return HasAllocationAt(instruction, /*index=*/{});
    [all...]

Completed in 1048 milliseconds