Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Doutfeed_thunk.cc48 TF_RETURN_IF_ERROR(outfeed_buffers->ForEachMutableElementWithStatus( in ExecuteOnStream()
Dgpu_executable.cc296 TF_RETURN_IF_ERROR(shaped_buffer.buffers().ForEachMutableElementWithStatus( in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree.h262 Status ForEachMutableElementWithStatus(const Fn& func);
614 Status ShapeTree<T>::ForEachMutableElementWithStatus(const Fn& func) { in ForEachMutableElementWithStatus() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc222 TF_RETURN_IF_ERROR(result_buffer.buffers().ForEachMutableElementWithStatus( in CreateResultShapedBuffer()