HomeSort by relevance Sort by last modified time
    Searched refs:StreamGroupFactory (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.h 111 class StreamGroupFactory;
gpu_device.cc 195 class BaseGPUDevice::StreamGroupFactory {
231 // Returns a reference to the StreamGroupFactory singleton. Note that this is
233 static StreamGroupFactory& Global() {
234 static StreamGroupFactory* instance = new StreamGroupFactory();
243 // StreamGroupFactory cannot be created directly; Call
244 // StreamGroupFactory::Global() to get the global instance.
245 StreamGroupFactory() = default;
246 TF_DISALLOW_COPY_AND_ASSIGN(StreamGroupFactory);
288 StreamGroupFactory::Global().GetOrCreate(tf_gpu_id_, i, executor_))
    [all...]

Completed in 188 milliseconds