Home
last modified time | relevance | path

Searched refs:input_alloc_attr (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc32 ctx, !ctx->input_alloc_attr(0).on_host(), in ComputeAsync()
65 ctx, ctx->input_alloc_attr(0).on_host(), in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc86 sub_input_attr_({ctx->input_alloc_attr(0), ctx->input_alloc_attr(0)}),
Dring_reducer.cc93 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
124 col_ctx_->op_ctx->input_alloc_attr(0))); in ContinueAfterInputCopy()
Dring_gatherer.cc105 col_ctx_->op_ctx->input_alloc_attr(0), in Run()
Dhierarchical_tree_broadcaster.cc372 col_ctx_->op_ctx->input_alloc_attr(0), in RunTree()
/external/tensorflow/tensorflow/core/kernels/
Dstack.cc236 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
263 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
Ddebug_ops.h76 !context->input_alloc_attr(0).on_host(); in Compute()
94 !context->input_alloc_attr(0).on_host(); in Compute()
Dregex_replace_op.cc40 ctx->input_memory_type(0), ctx->input_alloc_attr(0)); in InternalCompute()
Dcollective_nccl_reducer.cc108 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in Run()
Dsendrecv_ops.cc88 args.alloc_attrs = ctx->input_alloc_attr(0); in Compute()
Dunicode_ops.cc291 ctx->input_memory_type(0), ctx->input_alloc_attr(0)); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1041 AllocatorAttributes input_alloc_attr(int index) const { in input_alloc_attr() function
Dop_kernel.cc553 : input_alloc_attr(input_index); in forward_input()