Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device.cc70 class PluggableDevice::StreamGroupFactory { class in tensorflow::PluggableDevice
123 static StreamGroupFactory& Global() { in Global()
124 static StreamGroupFactory* instance = new StreamGroupFactory(); in Global()
135 StreamGroupFactory() = default;
136 TF_DISALLOW_COPY_AND_ASSIGN(StreamGroupFactory);
178 stream_ = StreamGroupFactory::Global().GetOrCreate( in Init()
Dpluggable_device.h91 class StreamGroupFactory; variable
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc241 class BaseGPUDevice::StreamGroupFactory { class in tensorflow::BaseGPUDevice
308 static StreamGroupFactory& Global() { in Global()
309 static StreamGroupFactory* instance = new StreamGroupFactory(); in Global()
383 StreamGroupFactory() = default;
384 TF_DISALLOW_COPY_AND_ASSIGN(StreamGroupFactory);
446 stream_ = StreamGroupFactory::Global().GetOrCreate( in Init()
2063 StreamGroupFactory::Global().TestOnlyReset(); in TestOnlyReset()
Dgpu_device.h170 class StreamGroupFactory; variable