Home
last modified time | relevance | path

Searched refs:output_alloc_attr (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dring_gatherer.cc91 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()
Dhierarchical_tree_broadcaster.cc373 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()
Dring_reducer.cc94 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()
Dring_alg.cc394 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()
Dexecutor.cc2026 out->alloc_attr = ctx->output_alloc_attr(i); in ProcessOutputs()
/external/tensorflow/tensorflow/core/kernels/
Dparse_tensor_op.cc52 proto, ctx->output_alloc_attr(0), &output)); in Compute()
Dsendrecv_ops.cc178 args.alloc_attrs = ctx->output_alloc_attr(0); in ComputeAsync()
Dscoped_allocator_ops.cc58 AllocatorAttributes attr = context->output_alloc_attr(0); in Compute()
Dreduction_ops_common.h173 const AllocatorAttributes alloc_attr = ctx->output_alloc_attr(0);
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc454 : output_alloc_attr(output_index); in forward_input_to_output_with_shape()
669 AllocatorAttributes attr = output_alloc_attr(index); in allocate_output()
Dop_kernel.h1051 AllocatorAttributes output_alloc_attr(int index) const { in output_alloc_attr() function