Home
last modified time | relevance | path

Searched refs:StreamGroup (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h138 struct StreamGroup { struct
146 gtl::InlinedVector<StreamGroup*, 4> streams_; argument
Dgpu_device.cc237 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id, in GetOrCreate()
242 StreamGroup* group = in GetOrCreate()
290 std::map<key_type, StreamGroup> streams_;