Searched refs:compute_stream (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 104 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local
107 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
136 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local
139 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
173 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local
176 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
208 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local
211 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
245 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local
248 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
|
| D | collective_nccl_reducer.cc | 124 auto* compute_stream = col_ctx_->op_ctx->op_device_context()->stream(); in Run() local
145 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_stream_util.cc | 45 (opts.compute_stream >= opts.max_streams)) { in AssignStreams()
99 if (opts.compute_stream >= 0) stream_id = opts.compute_stream; in AssignStreams()
|
| D | gpu_stream_util_test.cc | 51 opts.compute_stream = 5; in TEST_F()
124 opts.compute_stream = 93; in TEST_F()
|
| D | gpu_stream_util.h | 34 int32 compute_stream = -1; member
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 72 std::shared_ptr<se::Stream> compute_stream, in XlaDeviceContext() argument
79 : stream_(std::move(compute_stream)), in XlaDeviceContext()
|
| D | xla_device_context.h | 55 std::shared_ptr<se::Stream> compute_stream,
|