Home
last modified time | relevance | path

Searched refs:ForEachElement (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc53 ForEachElement( in IsAmbiguous()
63 ForEachElement([&](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsDistinct()
82 ForEachElement( in CreateFlattenedSet()
91 ForEachElement([&found, &buffer](const ShapeIndex& /*index*/, in ContainsBuffer()
188 points_to_set.ForEachElement(
349 operand_points_to_set.ForEachElement( in HandleCopyDone()
387 operand_points_to_set.ForEachElement( in HandleSend()
421 operand_points_to_set.ForEachElement( in HandleTuple()
586 .ForEachElement([buffers, instruction]( in GatherBuffersDefinedByInstruction()
660 points_to_set.ForEachElement([&prefix, &output]( in InstructionToString()
Dhlo_input_output_alias_config.cc56 alias_.ForEachElement( in ToProto()
127 alias_.ForEachElement( in ParameterMustAlias()
140 alias_.ForEachElement( in GetAliasedOutput()
159 alias_.ForEachElement( in ForEachAlias()
Dshaped_buffer_test.cc136 output.buffers().ForEachElement([&](const xla::ShapeIndex& sub_index, in TEST()
144 sb.buffers().ForEachElement( in TEST()
Dtuple_points_to_analysis.h125 void ForEachElement(const Fn& fn) const { in ForEachElement() function
126 tree_.ForEachElement([&fn](const ShapeIndex& index, const Elem& elem) { in ForEachElement()
Dhlo_value.cc298 ForEachElement([&out](const ShapeIndex& index, const HloValueSet& value_set) { in ToString()
Dhlo_liveness_analysis.cc55 index_tree.ForEachElement([&](const ShapeIndex& shape_index, bool live) { in ForEachLiveIndex()
Dexecutable.cc96 shaped_buffer.buffers().ForEachElement( in MakeMaybeOwningDeviceMemoryTree()
Dhlo_runner.cc133 input_buffer.buffers().ForEachElement( in ExecutionInputsFromScopedShapedBuffers()
Dhlo_memory_scheduler.cc122 points_to_analysis.GetPointsToSet(operand).ForEachElement( in ListScheduler()
Dlayout_assignment.cc114 points_to_analysis_.GetPointsToSet(inst).ForEachElement( in LayoutConstraints()
139 points_to_set.ForEachElement( in GetBufferSet()
Dhlo_dataflow_analysis.cc201 .ForEachElement([this, &instruction, &out]( in ToString()
Ddynamic_dimension_inference.cc1380 dynamic_output_mapping.ForEachElement( in HandleConditional()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree_test.cc65 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()
Dshape_tree.h249 void ForEachElement(const Fn& func) const;
269 ForEachElement([&](const ShapeIndex& index, const T& t) { in Map()
662 void ShapeTree<T>::ForEachElement(const Fn& func) const { in ForEachElement() function
724 ForEachElement([&other, &equal](const ShapeIndex& index, const T& data) {
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc380 ->buffers_.ForEachElement( in MakeSubBuffer()
516 elements.ForEachElement([&](const xla::ShapeIndex& index, in MakeTuple()
533 element.allocation->buffers_.ForEachElement( in MakeTuple()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc368 buffers.buffers().ForEachElement( in ~OutputBuffers()