| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter.cc | 91 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchSend()
92 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchSend()
93 col_ctx_->op_ctx->cancellation_manager(), done); in DispatchSend()
108 col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
109 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchRecv()
110 0, col_ctx_->op_ctx->cancellation_manager(), done); 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()
100 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
101 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in Run()
119 AllocatorAttributes attr = col_ctx_->op_ctx->output_alloc_attr(0); in ContinueAfterInputCopy()
141 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in ContinueAfterInputCopy()
143 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in ContinueAfterInputCopy()
258 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in RunAsyncParts()
273 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in RunAsyncParts()
|
| D | hierarchical_tree_broadcaster.cc | 376 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in RunTree()
379 col_ctx_->op_ctx->input_alloc_attr(0), in RunTree()
380 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in RunTree()
411 col_ctx_->op_ctx->op_kernel().name_view().data(), col_ctx_->step_id, in DispatchSend()
424 col_ctx_->op_ctx->op_device_context(), in DispatchSend()
425 col_ctx_->op_ctx->output_alloc_attr(0), src_tensor, in DispatchSend()
426 col_ctx_->device_locality, col_ctx_->op_ctx->cancellation_manager(), in DispatchSend()
445 col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
446 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
448 col_ctx_->op_ctx->cancellation_manager(), done); in DispatchRecv()
|
| D | ring_alg.cc | 256 col_ctx_->op_ctx->device()->tensorflow_gpu_device_info(); in TensorDebugString()
287 if (col_ctx_->op_ctx->cancellation_manager() == nullptr || in StartAbort()
288 (!col_ctx_->op_ctx->cancellation_manager()->IsCancelled() && in StartAbort()
289 !col_ctx_->op_ctx->cancellation_manager()->IsCancelling())) { in StartAbort()
395 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchSend()
396 col_ctx_->op_ctx->output_alloc_attr(0), &rf->chunk, in DispatchSend()
397 col_ctx_->device_locality, col_ctx_->op_ctx->cancellation_manager(), in DispatchSend()
416 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
417 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
419 col_ctx_->op_ctx->cancellation_manager(), done); in DispatchRecv()
|
| D | ring_gatherer.cc | 93 AllocatorAttributes attr = col_ctx_->op_ctx->output_alloc_attr(0); in Run()
107 col_ctx_->op_ctx->op_device_context(), in Run()
108 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, in Run()
109 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
110 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, &alias_chunk, in Run()
|
| D | collective_util.cc | 97 Status ComputeBinOp(OpKernelContext* op_ctx, OpKernelContext::Params* params, in ComputeBinOp() argument
106 new SubContext(op_ctx, params, op, output, input)); in ComputeBinOp()
|
| D | collective_util.h | 52 Status ComputeBinOp(OpKernelContext* op_ctx, OpKernelContext::Params* params,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 64 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in Run()
66 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in Run()
118 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in Run()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 78 auto* compute_stream = col_ctx->op_ctx->op_device_context()->stream(); in Enqueue()
79 auto* gpu_info = col_ctx->op_ctx->device()->tensorflow_gpu_device_info(); in Enqueue()
84 CancellationManager* cancel_mgr = col_ctx->op_ctx->cancellation_manager(); in Enqueue()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 114 OpKernelContext op_ctx(&op_params, 0); in MakeIterator() local
115 IteratorContext::Params params(&op_ctx); in MakeIterator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 177 op_ctx(ctx), in CollectiveContext()
|
| D | collective.h | 368 OpKernelContext* op_ctx; // Not owned member
|