Searched refs:hlo_use (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.cc | 960 absl::StrAppend(&out, " ", use.hlo_use.ToString(), "\n"); in ToString()
1090 allocation_value.uses().at(0).hlo_use.instruction->opcode())); in CreateAllocationValues()
1113 VLOG(3) << "Adding aliasing for use " << use->hlo_use.ToString() in FindAliases()
1124 maybe_add_alias_with_instruction(use.hlo_use.instruction, &use); in FindAliases()
1128 use.hlo_use.instruction->called_computations()) { in FindAliases()
1137 if (use.hlo_use.instruction->opcode() == HloOpcode::kWhile) { in FindAliases()
1139 use.hlo_use.instruction->while_body()->root_instruction(), in FindAliases()
1140 use.hlo_use.operand_index}; in FindAliases()
1142 << use.hlo_use.ToString() << " to " << root_alias; in FindAliases()
1233 if (other_use.hlo_use.instruction != use.instruction) { in IsUseAllowedInAlternateMemory()
[all …]
|
| D | memory_space_assignment.h | 831 HloUse hlo_use; member
839 return hlo_use == other.hlo_use && time == other.time &&
845 return H::combine(std::move(h), s.hlo_use, s.time, s.aliases); in AbslHashValue()
|
| D | hlo_instruction.cc | 3979 UseKind hlo_use = OperandElementUse(hlo, operand_num); in ComputeInternal() local
3982 if (hlo_use == UseKind::kNoUse) { in ComputeInternal()
3998 return std::min({old_val, hlo_use, operand_use}); in ComputeInternal()
|