Searched refs:op_ctx (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_reducer.cc | 99 col_ctx_->op_ctx->op_device_context(), in Run()
100 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, in Run()
101 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
102 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in Run()
120 AllocatorAttributes attr = col_ctx_->op_ctx->output_alloc_attr(0); in ContinueAfterInputCopy()
142 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in ContinueAfterInputCopy()
144 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in ContinueAfterInputCopy()
259 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in RunAsyncParts()
275 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in RunAsyncParts()
|
| D | ring_gatherer.cc | 94 AllocatorAttributes attr = col_ctx_->op_ctx->output_alloc_attr(0); in Run()
108 col_ctx_->op_ctx->op_device_context(), in Run()
109 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, in Run()
110 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
111 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, &alias_chunk, in Run()
|
| D | hierarchical_tree_broadcaster.cc | 378 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in RunTree()
381 col_ctx_->op_ctx->input_alloc_attr(0), in RunTree()
382 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in RunTree()
423 col_ctx_->op_ctx->op_device_context(), in DispatchSend()
424 col_ctx_->op_ctx->output_alloc_attr(0), in DispatchSend()
443 col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
444 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
|
| D | ring_alg.cc | 254 col_ctx_->op_ctx->device()->tensorflow_gpu_device_info(); in TensorDebugString()
393 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchSend()
394 col_ctx_->op_ctx->output_alloc_attr(0), &rf->chunk, in DispatchSend()
413 col_ctx_->device, col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
414 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
|
| 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 | 93 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in Run()
95 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in Run()
120 auto* compute_stream = col_ctx_->op_ctx->op_device_context()->stream(); in Run()
121 auto* gpu_info = col_ctx_->op_ctx->device()->tensorflow_gpu_device_info(); in Run()
210 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in Run()
|
| D | collective_nccl_gatherer.cc | 27 auto* compute_stream = col_ctx_->op_ctx->op_device_context()->stream(); in Run()
28 auto* gpu_info = col_ctx_->op_ctx->device()->tensorflow_gpu_device_info(); in Run()
|
| D | collective_nccl_broadcaster.cc | 27 auto* compute_stream = col_ctx_->op_ctx->op_device_context()->stream(); in Run()
28 auto* gpu_info = col_ctx_->op_ctx->device()->tensorflow_gpu_device_info(); in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 170 op_ctx(ctx), in CollectiveContext()
|
| D | collective.h | 366 OpKernelContext* op_ctx; // Not owned variable
|