Home
last modified time | relevance | path

Searched refs:allocation_value (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc797 AllocationValue* allocation_value = &allocation_values.at(i); in CreateAllocationValues() local
798 if (allocation_value->computation() == use_computation && in CreateAllocationValues()
800 allocation_value->defining_position().instruction) < use_time) { in CreateAllocationValues()
801 last_allocation_value = allocation_value; in CreateAllocationValues()
1298 for (AllocationValue& allocation_value : allocation_values) { in AllocateAllocationValues()
1300 instruction_schedule.at(allocation_value.defining_instruction()); in AllocateAllocationValues()
1304 preferred_offset_for_computation.find(allocation_value.computation()); in AllocateAllocationValues()
1310 for (int use_idx = 0; use_idx < allocation_value.uses().size(); ++use_idx) { in AllocateAllocationValues()
1311 const AllocationValue::Use& use = allocation_value.uses().at(use_idx); in AllocateAllocationValues()
1376 if (!IsUseAllowedInAlternateMemory(allocation_value, hlo_use)) { in AllocateAllocationValues()
[all …]
Dmemory_space_assignment.h1047 MemorySpaceAssignment::AllocationValue* allocation_value; member