Searched refs:ForEachElement (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 50 ForEachElement( in IsAmbiguous()
60 ForEachElement([&](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsDistinct()
79 ForEachElement( in CreateFlattenedSet()
88 ForEachElement([&found, &buffer](const ShapeIndex& /*index*/, in ContainsBuffer()
185 points_to_set.ForEachElement(
337 operand_points_to_set.ForEachElement( in HandleSend()
371 operand_points_to_set.ForEachElement( in HandleTuple()
506 .ForEachElement([buffers, instruction]( in GatherBuffersDefinedByInstruction()
580 points_to_set.ForEachElement([&prefix, &output]( in InstructionToString()
|
| D | hlo_input_output_alias_config.cc | 56 alias_.ForEachElement( in ToProto()
123 alias_.ForEachElement( in ParameterAliasKind()
136 alias_.ForEachElement( in GetAliasedOutput()
154 alias_.ForEachElement( in ForEachAlias()
|
| D | shaped_buffer_test.cc | 133 output.buffers().ForEachElement([&](const xla::ShapeIndex& sub_index, in TEST()
141 sb.buffers().ForEachElement( in TEST()
|
| D | tuple_points_to_analysis.h | 124 void ForEachElement(const Fn& fn) const { in ForEachElement() function
125 tree_.ForEachElement([&fn](const ShapeIndex& index, const Elem& elem) { in ForEachElement()
|
| D | hlo_value.cc | 285 ForEachElement([&out](const ShapeIndex& index, const HloValueSet& value_set) { in ToString()
|
| D | hlo_liveness_analysis.cc | 55 index_tree.ForEachElement([&](const ShapeIndex& shape_index, bool live) { in ForEachLiveIndex()
|
| D | hlo_memory_scheduler.cc | 119 points_to_analysis.GetPointsToSet(operand).ForEachElement( in ListScheduler()
|
| D | layout_assignment.cc | 109 points_to_analysis_.GetPointsToSet(inst).ForEachElement( in LayoutConstraints()
134 points_to_set.ForEachElement( in GetBufferSet()
|
| D | hlo_dataflow_analysis.cc | 155 .ForEachElement([this, &instruction, &out]( in ToString()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree_test.cc | 65 int_tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestShapeConstructor()
73 bool_tree.ForEachElement( in TestShapeConstructor()
91 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestInitValueConstructor()
107 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestInitValueConstructor()
163 shape_tree.ForEachElement( in TEST_F()
383 t.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TEST_F()
|
| D | shape_tree.h | 248 void ForEachElement(const Fn& func) const;
268 ForEachElement([&](const ShapeIndex& index, const T& t) { in Map()
620 void ShapeTree<T>::ForEachElement(const Fn& func) const { in ForEachElement() function
670 ForEachElement([&other, &equal](const ShapeIndex& index, const T& data) {
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 313 ->buffers_.ForEachElement( in MakeSubBuffer()
442 elements.ForEachElement([&](const xla::ShapeIndex& index, in MakeTuple()
461 element.allocation->buffers_.ForEachElement( in MakeTuple()
|