/external/tensorflow/tensorflow/compiler/xla/service/ |
hlo_value.cc | 250 std::ostream& operator<<(std::ostream& out, const HloValueSet& value_set) { 251 out << value_set.ToString(); 264 HloValueSet& value_set = pair.second; local 270 changed |= value_set.AssignUnionOf(input_value_sets); 285 ForEachElement([&out](const ShapeIndex& index, const HloValueSet& value_set) { 286 StrAppend(&out, " ", index.ToString(), " : ", value_set.ToString(), "\n");
|
bfloat16_propagation.cc | 214 auto& value_set = dataflow_->GetValueSet(&hlo, index); local 215 for (const HloValue* value : value_set.values()) { [all...] |
hlo_alias_analysis.cc | 379 const HloValueSet& value_set = pair.second; local 381 for (const HloValue* value : value_set.values()) { 397 const HloValueSet& value_set = pair.second; local 398 if (value_set.values().size() == 1) { 400 .insert(&GetBufferContainingValue(value_set.GetUniqueValue())) 412 for (const HloValue* value : value_set.values()) { 513 auto& value_set = buffer_map.GetValuesInBuffer(buffer_number); local 514 std::vector<const HloValue*> sorted_values(value_set.begin(), 515 value_set.end());
|
copy_insertion.cc | 1058 const HloValueSet& value_set = pair.second; local 1261 const HloValueSet& value_set = pair.second; local [all...] |
hlo_dataflow_analysis.cc | 80 const HloValueSet& value_set = GetValueSet(instruction, index); local 81 if (value_set.values().size() != 1) { 84 return value_set.GetUniqueValue().defining_instruction() == instruction; 129 const HloValueSet& value_set = index_value_set.second; local 130 for (const HloValue* value : value_set.values()) { 157 const HloValueSet& value_set) { 159 for (const HloValue* value : value_set.values()) { 199 HloValueSet& value_set = pair.second; local 203 CHECK_LE(value_set.values().size(), 1); 205 value_set.values().size() == 1 ? value_set.values()[0] : nullptr 246 << " previously had non-empty value set. Value set: " << value_set; local 331 HloValueSet& value_set = GetValueSet(send, index); local 346 HloValueSet& value_set = pair.second; local 364 InstructionValueSet& value_set = GetInstructionValueSet(call); local 400 HloValueSet& value_set = pair.second; local 418 HloValueSet& value_set = pair.second; local 450 HloValueSet& value_set = pair.second; local 553 HloValueSet& value_set = pair.second; local 575 HloValueSet& value_set = GetValueSet(tuple, index); local 863 const HloValueSet& value_set = pair.second; local 898 const HloValueSet& value_set = GetValueSet(position); local 911 const HloValueSet& value_set = pair.second; local [all...] |