Searched refs:ReleaseCurrentBuffer (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | xfeed_manager.h | 72 // buffer, i.e., ReleaseCurrentBuffer must be called between calls to 86 void ReleaseCurrentBuffer(int32 length, void* data, StatusOr<Shape> shape);
|
H A D | cpu_runtime.cc | 109 xfeed->infeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, 141 xfeed->outfeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr,
|
H A D | xfeed_manager.cc | 70 void XfeedQueueManager::ReleaseCurrentBuffer(int32 length, void* data, function in class:xla::cpu::runtime::XfeedQueueManager
|
Completed in 213 milliseconds