Home
last modified time | relevance | path

Searched refs:ForEachMutableElementWithStatus (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_transfer_manager.cc100 TF_RETURN_IF_ERROR(device_buffer->buffers().ForEachMutableElementWithStatus( in ReadDynamicShapes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_xfeed.cc281 TF_RETURN_IF_ERROR(device_buffer->buffers().ForEachMutableElementWithStatus( in ReadDynamicShapesOnCpu()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_tree.h240 Status ForEachMutableElementWithStatus( in ForEachMutableElementWithStatus() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtransfer_manager.cc206 TF_RETURN_IF_ERROR(device_buffer->buffers().ForEachMutableElementWithStatus( in ReadDynamicShapes()