Searched refs:output_alloc_attr (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/ |
D | ring_gatherer.cc | 91 AllocatorAttributes attr = col_ctx_->op_ctx->output_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 | 373 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in RunTree() 414 col_ctx_->op_ctx->output_alloc_attr(0), in DispatchSend() 434 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
|
D | ring_reducer.cc | 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()
|
D | ring_alg.cc | 394 col_ctx_->op_ctx->output_alloc_attr(0), &rf->chunk, in DispatchSend() 414 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
|
D | executor.cc | 2026 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 | sendrecv_ops.cc | 178 args.alloc_attrs = ctx->output_alloc_attr(0); in ComputeAsync()
|
D | scoped_allocator_ops.cc | 58 AllocatorAttributes attr = context->output_alloc_attr(0); in Compute()
|
D | reduction_ops_common.h | 173 const AllocatorAttributes alloc_attr = ctx->output_alloc_attr(0);
|
/external/tensorflow/tensorflow/core/framework/ |
D | op_kernel.cc | 454 : output_alloc_attr(output_index); in forward_input_to_output_with_shape() 669 AllocatorAttributes attr = output_alloc_attr(index); in allocate_output()
|
D | op_kernel.h | 1051 AllocatorAttributes output_alloc_attr(int index) const { in output_alloc_attr() function
|