Searched refs:HasAllocationAt (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 84 return assignment.HasAllocationAt(operand, index) && in CanEmitFusedDynamicUpdateSliceInPlace()
85 assignment.HasAllocationAt(fusion, {}) && in CanEmitFusedDynamicUpdateSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 407 bool HasAllocationAt(const HloInstruction* instruction,
|
| D | buffer_assignment_test.cc | 386 TEST_F(BufferAssignmentTest, HasAllocationAt) { in TEST_F() argument
405 buffers->HasAllocationAt(tuple, /*index=*/{})); in TEST_F()
407 buffers->HasAllocationAt(tuple, /*index=*/{0})); in TEST_F()
409 buffers->HasAllocationAt(tuple, /*index=*/{1})); in TEST_F()
411 buffers->HasAllocationAt(tuple, /*index=*/{2})); in TEST_F()
|
| D | buffer_assignment.cc | 456 bool BufferAssignment::HasAllocationAt(const HloInstruction* instruction, in HasAllocationAt() function in xla::BufferAssignment
469 return HasAllocationAt(instruction, /*index=*/{}); in HasTopLevelAllocation()
|