Searched refs:AddRequiredAssignment (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.cc | 1263 AddRequiredAssignment(position.instruction, position.index, in AddRequiredAssignmentsForColocatedIntervals()
1267 AddRequiredAssignment(called_computation->root_instruction(), in AddRequiredAssignmentsForColocatedIntervals()
1377 AddRequiredAssignment(allocation_value.value(), hlo_use.instruction, in AllocateAllocationValues()
1681 AddRequiredAssignment(instruction, index, aliased_allocation->memory_space(), in AddAliasedRequiredAssignment()
1685 void AlternateMemoryBestFitHeap::AddRequiredAssignment( in AddRequiredAssignment() function in xla::AlternateMemoryBestFitHeap
1710 void AlternateMemoryBestFitHeap::AddRequiredAssignment( in AddRequiredAssignment() function in xla::AlternateMemoryBestFitHeap
1717 AddRequiredAssignment(value, instruction, memory_space, instruction_time, in AddRequiredAssignment()
|
| D | memory_space_assignment.h | 1220 void AddRequiredAssignment(const HloValue* value,
1224 void AddRequiredAssignment(const HloInstruction* instruction,
|