Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched defs:compute_stream (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_stream_util.h	34 int32 compute_stream = -1; member in struct:tensorflow::gpu_stream_util::AssignStreamsOpts
/external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_ops.cc	103 auto* compute_stream = c->op_device_context()->stream(); variable 107 compute_stream->parent(), gpu_info->gpu_id, gpu_info->event_mgr, 108 compute_stream, in_t, out_t, std::move(actual_done)); variable 128 auto* compute_stream = c->op_device_context()->stream(); variable 132 compute_stream->parent(), gpu_info->gpu_id, gpu_info->event_mgr, 133 compute_stream, &c->input(0), std::move(actual_done)); variable 157 auto* compute_stream = c->op_device_context()->stream(); variable 161 compute_stream->parent(), gpu_info->gpu_id, gpu_info->event_mgr, 162 compute_stream, &in_t, out_t, std::move(actual_done)); variable 185 auto* compute_stream = c->op_device_context()->stream() variable 189 gpu_info->gpu_id, gpu_info->event_mgr, compute_stream, &c->input(0), variable 217 auto* compute_stream = c->op_device_context()->stream(); variable 221 gpu_info->gpu_id, gpu_info->event_mgr, compute_stream, out_t, variable     [all...]

Completed in 185 milliseconds