OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:InfeedManager
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
infeed_manager.cc
27
InfeedManager
::
InfeedManager
() : host_to_device_executor_(nullptr) {}
29
void
InfeedManager
::Reset() {
38
void
InfeedManager
::EnqueueBuffers(const std::vector<InfeedBuffer*>& buffers) {
53
InfeedBuffer*
InfeedManager
::BlockingDequeueBuffer() {
64
void
InfeedManager
::ReleaseBuffers(const std::vector<InfeedBuffer*>& buffers) {
77
se::Stream*
InfeedManager
::GetStream(se::StreamExecutor* executor) {
93
InfeedManager
* GetOrCreateInfeedManager() {
94
static
InfeedManager
* manager = new
InfeedManager
;
[
all
...]
infeed_manager.h
74
class
InfeedManager
{
76
InfeedManager
();
87
//
InfeedManager
, either as a result of a call to Reset or because the
131
InfeedManager
* GetOrCreateInfeedManager();
gpu_transfer_manager.cc
102
gpu::
InfeedManager
* infeed_manager = gpu::GetOrCreateInfeedManager();
135
gpu::
InfeedManager
* infeed_manager = gpu::GetOrCreateInfeedManager();
infeed_thunk.cc
40
InfeedManager
* infeed_manager = GetOrCreateInfeedManager();
Completed in 856 milliseconds