Searched refs:output_alloc_attr (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | all_to_all.cc | 72 col_ctx_->op_ctx->output_alloc_attr(0), &output_buffer_, in CheckCounterAndCallDone()
101 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->output_alloc_attr(0)), in Run()
131 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchSend()
144 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchRecv()
|
| 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 | 381 col_ctx_->op_ctx->output_alloc_attr(0), col_ctx_->input, in RunTree()
428 col_ctx_->op_ctx->output_alloc_attr(0), src_tensor, in DispatchSend()
449 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 | 411 col_ctx_->op_ctx->output_alloc_attr(0), &rf->chunk, in DispatchSend()
432 col_ctx_->op_ctx->output_alloc_attr(0), dst_tensor, in DispatchRecv()
|
| D | single_threaded_executor.cc | 204 kernel_state.output_alloc_attr.set_on_host(on_host); in Initialize()
561 AllocatorAttributes output_alloc_attr; member
|
| D | executor.cc | 1074 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 | reduction_ops_common.h | 165 const AllocatorAttributes alloc_attr = ctx->output_alloc_attr(0);
|
| D | sendrecv_ops.cc | 204 args.alloc_attrs = ctx->output_alloc_attr(0); in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 483 : output_alloc_attr(output_index); in forward_input_to_output_with_shape()
717 AllocatorAttributes attr = output_alloc_attr(index); in allocate_output()
920 if (allocated_scope_ids_->find(output_alloc_attr(index).scope_id) == in maybe_set_output_by_allocate_and_copy()
931 << output_alloc_attr(index).scope_id; in maybe_set_output_by_allocate_and_copy()
943 << output_alloc_attr(index).scope_id; in maybe_set_output_by_allocate_and_copy()
949 output_alloc_attr(index)); in maybe_set_output_by_allocate_and_copy()
|
| D | op_kernel.h | 1042 AllocatorAttributes output_alloc_attr(int index) const { in output_alloc_attr() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_framework_c_interface.cc | 81 ctx->output_alloc_attr(output_index)); in _mlir_ciface_tf_alloc()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 145 args.alloc_attrs = ctx->output_alloc_attr(i); in ComputeAsync()
|