Searched refs:op_ctx (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_reducer.cc | 91 col_ctx_->op_ctx->input_device_context(0), in Run()
92 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, in Run()
93 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
94 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in Run()
112 AllocatorAttributes attr = col_ctx_->op_ctx->output_alloc_attr(0); in ContinueAfterInputCopy()
124 col_ctx_->op_ctx->input_alloc_attr(0))); in ContinueAfterInputCopy()
125 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in ContinueAfterInputCopy()
235 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in RunAsyncParts()
250 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in RunAsyncParts()
|
| D | ring_gatherer.cc | 91 AllocatorAttributes attr = col_ctx_->op_ctx->output_alloc_attr(0); in Run()
103 col_ctx_->op_ctx->input_device_context(0), in Run()
104 col_ctx_->op_ctx->op_device_context(), col_ctx_->device, col_ctx_->device, in Run()
105 col_ctx_->op_ctx->input_alloc_attr(0), in Run()
106 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, &alias_chunk, in Run()
|
| D | hierarchical_tree_broadcaster.cc | 369 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in RunTree()
372 col_ctx_->op_ctx->input_alloc_attr(0), in RunTree()
373 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in RunTree()
413 col_ctx_->op_ctx->op_device_context(), in DispatchSend()
414 col_ctx_->op_ctx->output_alloc_attr(0), in DispatchSend()
433 col_ctx_->op_ctx->op_device_context(), in DispatchRecv()
434 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 | 99 Status ComputeBinOp(OpKernelContext* op_ctx, OpKernelContext::Params* params, in ComputeBinOp() argument
108 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 | 108 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in Run()
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()
125 auto* gpu_info = col_ctx_->op_ctx->device()->tensorflow_gpu_device_info(); in Run()
196 col_ctx_->op_ctx, col_ctx_->op_params, col_ctx_->device, in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 96 OpKernelContext op_ctx(&op_params, 0); in MakeIterator() local
97 IteratorContext::Params params(&op_ctx); in MakeIterator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 165 op_ctx(ctx), in CollectiveContext()
|
| D | collective.h | 350 OpKernelContext* op_ctx; // Not owned variable
|