Searched refs:output_alloc_attr (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter.cc | 92 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchSend()
109 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchRecv()
|
| D | ring_gatherer.cc | 93 AllocatorAttributes attr = col_ctx_->op_ctx->output_alloc_attr(0); in Run()
110 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, &alias_chunk, in Run()
|
| D | hierarchical_tree_broadcaster.cc | 380 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in RunTree()
425 col_ctx_->op_ctx->output_alloc_attr(0), src_tensor, in DispatchSend()
446 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
|
| D | ring_reducer.cc | 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()
|
| D | ring_alg.cc | 396 col_ctx_->op_ctx->output_alloc_attr(0), &rf->chunk, in DispatchSend()
417 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
|
| D | executor.cc | 1012 out->alloc_attr = ctx->output_alloc_attr(i); in ProcessOutputs()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | parse_tensor_op.cc | 52 proto, ctx->output_alloc_attr(0), &output)); in Compute()
|
| D | scoped_allocator_ops.cc | 58 AllocatorAttributes attr = context->output_alloc_attr(0); in Compute()
|
| D | sendrecv_ops.cc | 198 args.alloc_attrs = ctx->output_alloc_attr(0); in ComputeAsync()
|
| D | reduction_ops_common.h | 166 const AllocatorAttributes alloc_attr = ctx->output_alloc_attr(0);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 190 kernel_state.output_alloc_attr.set_on_host(on_host); in Initialize()
517 AllocatorAttributes output_alloc_attr; member
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 470 : output_alloc_attr(output_index); in forward_input_to_output_with_shape()
673 AllocatorAttributes attr = output_alloc_attr(index); in allocate_output()
909 if (allocated_scope_ids_->find(output_alloc_attr(index).scope_id) == in maybe_set_output_by_allocate_and_copy()
920 << output_alloc_attr(index).scope_id; in maybe_set_output_by_allocate_and_copy()
932 << output_alloc_attr(index).scope_id; in maybe_set_output_by_allocate_and_copy()
938 output_alloc_attr(index)); in maybe_set_output_by_allocate_and_copy()
|
| D | op_kernel.h | 1086 AllocatorAttributes output_alloc_attr(int index) const { in output_alloc_attr() function
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 145 args.alloc_attrs = ctx->output_alloc_attr(i); in ComputeAsync()
|