Searched refs:input_alloc_attr (Results 1 – 13 of 13) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | gpu_swapping_kernels.cc | 32 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/ |
D | collective_util.cc | 86 sub_input_attr_({ctx->input_alloc_attr(0), ctx->input_alloc_attr(0)}),
|
D | ring_reducer.cc | 93 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run() 124 col_ctx_->op_ctx->input_alloc_attr(0))); in ContinueAfterInputCopy()
|
D | ring_gatherer.cc | 105 col_ctx_->op_ctx->input_alloc_attr(0), in Run()
|
D | hierarchical_tree_broadcaster.cc | 372 col_ctx_->op_ctx->input_alloc_attr(0), in RunTree()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | stack.cc | 236 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync() 263 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
|
D | debug_ops.h | 76 !context->input_alloc_attr(0).on_host(); in Compute() 94 !context->input_alloc_attr(0).on_host(); in Compute()
|
D | regex_replace_op.cc | 40 ctx->input_memory_type(0), ctx->input_alloc_attr(0)); in InternalCompute()
|
D | collective_nccl_reducer.cc | 108 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in Run()
|
D | sendrecv_ops.cc | 88 args.alloc_attrs = ctx->input_alloc_attr(0); in Compute()
|
D | unicode_ops.cc | 291 ctx->input_memory_type(0), ctx->input_alloc_attr(0)); in Compute()
|
/external/tensorflow/tensorflow/core/framework/ |
D | op_kernel.h | 1041 AllocatorAttributes input_alloc_attr(int index) const { in input_alloc_attr() function
|
D | op_kernel.cc | 553 : input_alloc_attr(input_index); in forward_input()
|