| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 54 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr, in Run()
78 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in Run()
132 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr, in Compile()
|
| D | xla_launch_util.cc | 260 CHECK(ctx->op_device_context() && ctx->op_device_context()->stream()) in PopulateInputs()
265 ctx->op_device_context()->stream()); in PopulateInputs()
364 ctx->op_device_context()->CopyCPUTensorToDevice( in SetOutputForConstant()
374 static_cast<GPUDeviceContext*>(ctx->op_device_context()); in SetOutputForConstant()
428 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in PopulateOutputs()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 108 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
139 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
175 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
209 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
245 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync()
|
| D | debug_ops.h | 89 DeviceContext* device_ctxt = context->op_device_context(); in Compute()
733 auto* stream = context->op_device_context()->stream();
760 auto* stream = context->op_device_context()->stream();
787 auto* stream = context->op_device_context()->stream();
819 auto* stream = context->op_device_context()->stream();
852 auto* stream = context->op_device_context()->stream();
|
| D | collective_nccl_test.cc | 295 params->op_device_context = dev_ctx; in PrepareDeviceContext()
346 op_params.op_device_context->Unref(); in RunReduce()
379 op_params.op_device_context->Unref(); in RunBroadcast()
417 op_params.op_device_context->Unref(); in RunGather()
|
| D | gpu_device_array.h | 81 auto stream = context_->op_device_context()->stream(); in Finalize()
|
| D | sendrecv_ops.cc | 90 args.device_context = ctx->op_device_context(); in Compute()
197 args.device_context = ctx->op_device_context(); in ComputeAsync()
|
| D | collective_nccl_reducer.cc | 66 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in Run()
|
| D | stack.cc | 245 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync()
302 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync()
|
| D | cudnn_pooling_gpu.cc | 98 auto* stream = context->op_device_context()->stream(); in Compute()
240 auto* stream = context->op_device_context()->stream(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 196 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr, in CompileToLocalExecutable()
248 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in Compute()
253 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr, in Compute()
469 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr, in Compute()
472 ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; in Compute()
|
| /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/core/util/ |
| D | rocm_solvers.cc | 137 context->op_device_context()->stream()->parent()->implementation()); in ROCmSolver()
139 reinterpret_cast<const hipStream_t*>(context->op_device_context() in ROCmSolver()
235 context_->op_device_context()->stream()->parent()->implementation()); \
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_util.cc | 90 sub_params_.op_device_context = ctx->op_device_context();
|
| D | permuter.cc | 91 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchSend()
108 col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
|
| D | ring_reducer.cc | 98 col_ctx_->op_ctx->op_device_context(), in Run()
99 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, in Run()
143 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in ContinueAfterInputCopy()
|
| D | ring_gatherer.cc | 107 col_ctx_->op_ctx->op_device_context(), in Run()
108 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, in Run()
|
| D | hierarchical_tree_broadcaster.cc | 376 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in RunTree()
424 col_ctx_->op_ctx->op_device_context(), in DispatchSend()
445 col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | blas_gemm.cc | 53 ctx->op_device_context() in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 471 se::Stream* stream = context->op_device_context() in DoWork()
472 ? context->op_device_context()->stream() in DoWork()
554 se::Stream* stream = context->op_device_context() in DoWork()
555 ? context->op_device_context()->stream() in DoWork()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_rocm_runtime_wrappers.cc | 104 stream_executor::Stream *se_stream = ctx->op_device_context()->stream(); in tfKernelGenLaunchKernel()
|
| D | tf_cuda_runtime_wrappers.cc | 103 stream_executor::Stream *se_stream = ctx->op_device_context()->stream(); in tfKernelGenLaunchKernel()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 144 args.device_context = ctx->op_device_context(); in ComputeAsync()
271 args.device_context = ctx->op_device_context(); in Compute()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 206 if (cc_ctx->op_device_context() == nullptr) { // CPU Device in TF_GetStream()
210 } else if (!cc_ctx->op_device_context()->IsPluggableDevice()) { in TF_GetStream()
217 cc_ctx->op_device_context()->stream()->implementation()); in TF_GetStream()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 678 DeviceContext* op_device_context = nullptr; member
1066 T* op_device_context();
1067 DeviceContext* op_device_context() { in op_device_context() function
1068 DeviceContext* ret = params_->op_device_context; in op_device_context()
1668 T* OpKernelContext::op_device_context() { in op_device_context() function
1671 return static_cast<T*>(op_device_context()); in op_device_context()
|