| /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...] |
