| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 104 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
136 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
173 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
208 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
245 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
|
| D | collective_nccl_reducer.cc | 110 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in Run()
124 auto* compute_stream = col_ctx_->op_ctx->op_device_context()->stream(); in Run()
|
| D | sendrecv_ops.cc | 87 args.device_context = ctx->op_device_context(); in Compute()
177 args.device_context = ctx->op_device_context(); in ComputeAsync()
|
| D | cudnn_pooling_gpu.cc | 98 auto* stream = context->op_device_context()->stream(); in Compute()
225 auto* stream = context->op_device_context()->stream(); in Compute()
|
| D | gpu_device_array.h | 79 auto stream = context_->op_device_context()->stream(); in Finalize()
|
| D | stack.cc | 244 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync()
301 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync()
|
| D | where_op.cc | 293 auto stream = context->op_device_context()->stream(); in ComputeAsyncType()
306 auto stream = context->op_device_context()->stream(); in ComputeAsyncType()
|
| D | cudnn_rnn_ops.cc | 656 StreamExecutor* executor = context->op_device_context()->stream()->parent(); in CreateForwardAndBackwardIODescriptors()
769 Stream* stream = context->op_device_context()->stream(); in DoForward()
847 Stream* stream = context->op_device_context()->stream(); in DoBackward()
960 Stream* stream = context->op_device_context()->stream(); in ExtractCudnnRNNParamsInfo()
982 StreamExecutor* executor = context->op_device_context()->stream()->parent(); in CreateRnnDescriptor()
1083 Stream* stream = context->op_device_context()->stream(); in Compute()
1223 Stream* stream = context->op_device_context()->stream(); in Compute()
1460 auto* stream = context->op_device_context()->stream(); in MaybeAutoTune()
|
| D | pooling_ops_common.cc | 262 auto* stream = context->op_device_context()->stream(); in Compute()
413 auto* stream = context->op_device_context()->stream(); in Compute()
|
| D | check_numerics_op.cc | 142 auto* stream = context->op_device_context()->stream(); in ComputeAsync()
|
| D | matrix_triangular_solve_op.cc | 188 auto* stream = context->op_device_context()->stream(); in ComputeMatrix()
|
| D | lrn_op.cc | 210 auto* stream = context->op_device_context()->stream(); in launch()
429 auto* stream = context->op_device_context()->stream(); in launch()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 45 ctx->op_device_context()->CopyDeviceTensorToCPU( in ComputeAsync()
74 ctx->op_device_context()->CopyCPUTensorToDevice( in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 284 if (ctx->op_device_context() != nullptr) { in CompileToLocalExecutable()
286 ctx->op_device_context()->stream()->parent()->device_ordinal(); in CompileToLocalExecutable()
347 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in Compute()
533 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 212 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in PopulateInputs()
264 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in PopulateOutputs()
322 ctx->op_device_context()->CopyCPUTensorToDevice( in PopulateOutputs()
332 static_cast<GPUDeviceContext*>(ctx->op_device_context()); in PopulateOutputs()
|
| D | xla_compile_on_demand_op.cc | 68 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in Run()
146 ctx->op_device_context()->CopyDeviceTensorToCPU( in Compile()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.cc | 369 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in RunTree()
413 col_ctx_->op_ctx->op_device_context(), in DispatchSend()
433 col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
|
| D | ring_reducer.cc | 92 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, in Run()
125 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in ContinueAfterInputCopy()
|
| D | ring_alg.cc | 393 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchSend()
413 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
|
| D | ring_gatherer.cc | 104 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, col_ctx_->device, in Run()
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | blas_gemm.cc | 53 ctx->op_device_context() in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 201 se::Stream* stream = context->op_device_context() in DoWork()
202 ? context->op_device_context()->stream() in DoWork()
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | inc_op_kernel.cu.cc | 67 reinterpret_cast<const cudaStream_t*>(context->op_device_context() in Compute()
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| D | adjust_hsv_in_yiq_op_gpu.cu.cc | 70 auto* stream = ctx->op_device_context()->stream(); in operator ()()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 660 DeviceContext* op_device_context = nullptr; member
1031 T* op_device_context();
1032 DeviceContext* op_device_context() { in op_device_context() function
1033 DeviceContext* ret = params_->op_device_context; in op_device_context()
1611 T* OpKernelContext::op_device_context() { in op_device_context() function
1614 return static_cast<T*>(op_device_context()); in op_device_context()
|