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/
ops.h 58 return assignment.HasAllocationAt(operand, index) &&
59 assignment.HasAllocationAt(fusion, {}) &&
  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment.h 299 bool HasAllocationAt(const HloInstruction* instruction,
buffer_assignment_test.cc 308 TEST_F(BufferAssignmentTest, HasAllocationAt) {
310 // HasAllocationAt works correctly.
324 // Make sure that HasAllocationAt() agrees with what HasTopLevelAllocation()
327 buffers->HasAllocationAt(tuple, /*index=*/{}));
329 buffers->HasAllocationAt(tuple, /*index=*/{0}));
331 buffers->HasAllocationAt(tuple, /*index=*/{1}));
333 buffers->HasAllocationAt(tuple, /*index=*/{2}));
    [all...]
buffer_assignment.cc 212 bool BufferAssignment::HasAllocationAt(const HloInstruction* instruction,
225 return HasAllocationAt(instruction, /*index=*/{});
    [all...]

Completed in 74 milliseconds