Searched refs:input_alloc_attr (Results 1 – 16 of 16) 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 | 87 sub_input_attr_({ctx->input_alloc_attr(0), ctx->input_alloc_attr(0)}) {
|
| D | ring_reducer.cc | 100 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
141 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in ContinueAfterInputCopy()
|
| D | all_to_all.cc | 71 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in CheckCounterAndCallDone()
|
| D | ring_gatherer.cc | 109 col_ctx_->device, col_ctx_->op_ctx->input_alloc_attr(0), in Run()
|
| D | hierarchical_tree_broadcaster.cc | 380 col_ctx_->op_ctx->input_alloc_attr(0), in RunTree()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 237 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
264 AllocatorAttributes alloc_attrs = ctx->input_alloc_attr(1); in ComputeAsync()
|
| D | collective_nccl_reducer.cc | 64 col_ctx_->device->GetAllocator(col_ctx_->op_ctx->input_alloc_attr(0)), in Run()
|
| D | regex_replace_op.cc | 42 ctx->input_memory_type(0), ctx->input_alloc_attr(0)); in InternalCompute()
|
| D | sendrecv_ops.cc | 93 args.alloc_attrs = ctx->input_alloc_attr(0); in Compute()
|
| D | unicode_ops.cc | 293 ctx->input_memory_type(0), ctx->input_alloc_attr(0)); in Compute()
|
| D | debug_ops.h | 87 !context->input_alloc_attr(0).on_host(); in Compute()
|
| D | functional_ops.cc | 439 ctx->input_alloc_attr(i)); in GetArgsFromContext()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 272 args.alloc_attrs = ctx->input_alloc_attr(i); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 1032 AllocatorAttributes input_alloc_attr(int index) const { in input_alloc_attr() function
|
| D | op_kernel.cc | 568 : input_alloc_attr(input_index); in forward_input()
|