Searched refs:HasAllocationAt (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.h | 61 return assignment.HasAllocationAt(operand, index) && in CanEmitFusedDynamicUpdateSliceInPlace()
62 assignment.HasAllocationAt(fusion, {}) && in CanEmitFusedDynamicUpdateSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 387 bool HasAllocationAt(const HloInstruction* instruction,
|
| D | buffer_assignment_test.cc | 379 TEST_F(BufferAssignmentTest, HasAllocationAt) { in TEST_F() argument
398 buffers->HasAllocationAt(tuple, /*index=*/{})); in TEST_F()
400 buffers->HasAllocationAt(tuple, /*index=*/{0})); in TEST_F()
402 buffers->HasAllocationAt(tuple, /*index=*/{1})); in TEST_F()
404 buffers->HasAllocationAt(tuple, /*index=*/{2})); in TEST_F()
|
| D | buffer_assignment.cc | 394 bool BufferAssignment::HasAllocationAt(const HloInstruction* instruction, in HasAllocationAt() function in xla::BufferAssignment
407 return HasAllocationAt(instruction, /*index=*/{}); in HasTopLevelAllocation()
|