Searched refs:ForEachMutableElement (Results 1 – 12 of 12) sorted by relevance
206 points_to_set.ForEachMutableElement( in DefaultAction()234 points_to_set.ForEachMutableElement( in HandleGetTupleElement()306 points_to_set.ForEachMutableElement( in HandleRecvDone()326 points_to_set.ForEachMutableElement( in HandleCopyStart()457 points_to_set.ForEachMutableElement( in HandleTupleSelect()485 points_to_set.ForEachMutableElement([&](const ShapeIndex& index, in HandleCustomCall()615 dst_points_to_set.ForEachMutableElement( in CreateCopiedPointsToSet()
123 sb.buffers().ForEachMutableElement( in TEST()160 sb.buffers().ForEachMutableElement( in TEST()
131 void ForEachMutableElement(const Fn& fn) { in ForEachMutableElement() function132 tree_.ForEachMutableElement([&fn](const ShapeIndex& index, Elem* elem) { in ForEachMutableElement()
179 it->second.ForEachMutableElement( in ComputeHloReplicationOnComputation()301 shape_tree.ForEachMutableElement( in ComputeHloReplicationOnComputation()
1084 instruction_value_set.ForEachMutableElement( in OptimizePhiValues()
256 void ForEachMutableElement(const Fn& func);278 ForEachMutableElement([&](const ShapeIndex& index, T* t) { in Map()674 void ShapeTree<T>::ForEachMutableElement(const Fn& func) { in ForEachMutableElement() function693 ForEachMutableElement([this, &other, &source_base_index, &target_base_index]( in CopySubtreeFrom()
98 tree.ForEachMutableElement( in TestInitValueConstructor()175 shape_tree.ForEachMutableElement( in TEST_F()614 shape_tree.ForEachMutableElement([](const ShapeIndex& index, int* data) { in BM_ForEach()
387 ->buffers_.ForEachMutableElement( in MakeSubBuffer()527 element.allocation->buffers_.ForEachMutableElement( in MakeTuple()
295 buffers->buffers().ForEachMutableElement([&](const xla::ShapeIndex& index, in BuildComputationInputs()469 shaped_buffer.buffers().ForEachMutableElement( in AllocateOutputTensors()
111 owned_buffers.ForEachMutableElement( in RunAndCheck()
278 xla_tensor->shaped_buffer().buffers().ForEachMutableElement( in PopulateInputs()
141 tuple_shape_tree->ForEachMutableElement( in ParseTupleTree()