Home
last modified time | relevance | path

Searched refs:ForEachMutableElement (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc203 points_to_set.ForEachMutableElement( in DefaultAction()
231 points_to_set.ForEachMutableElement( in HandleGetTupleElement()
303 points_to_set.ForEachMutableElement( in HandleRecvDone()
407 points_to_set.ForEachMutableElement( in HandleTupleSelect()
535 dst_points_to_set.ForEachMutableElement( in CreateCopiedPointsToSet()
Dtuple_points_to_analysis.h130 void ForEachMutableElement(const Fn& fn) { in ForEachMutableElement() function
131 tree_.ForEachMutableElement([&fn](const ShapeIndex& index, Elem* elem) { in ForEachMutableElement()
Dshaped_buffer_test.cc120 sb.buffers().ForEachMutableElement( in TEST()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree.h255 void ForEachMutableElement(const Fn& func);
277 ForEachMutableElement([&](const ShapeIndex& index, T* t) { in Map()
632 void ShapeTree<T>::ForEachMutableElement(const Fn& func) { in ForEachMutableElement() function
649 ForEachMutableElement([this, &other, &source_base_index, &target_base_index]( in CopySubtreeFrom()
Dshape_tree_test.cc98 tree.ForEachMutableElement( in TestInitValueConstructor()
175 shape_tree.ForEachMutableElement( in TEST_F()
585 shape_tree.ForEachMutableElement([](const ShapeIndex& index, int* data) { in BM_ForEach()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.cc154 outfeed_buffers.ForEachMutableElement( in TransferLiteralFromOutfeed()
Dir_emitter_unnested.cc1733 slices.ForEachMutableElement( in BuildInfeedThunk()
1747 slices.ForEachMutableElement( in BuildOutfeedThunk()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc320 ->buffers_.ForEachMutableElement( in MakeSubBuffer()
453 element.allocation->buffers_.ForEachMutableElement( in MakeTuple()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h137 tuple_shape_tree->ForEachMutableElement( in ParseTupleTree()