Searched refs:alloc_attr (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils.cc | 102 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
105 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
107 arg_alloc_attrs->push_back(alloc_attr); in UpdateArgAndRetvalMetadata()
113 AllocatorAttributes alloc_attr; in UpdateArgAndRetvalMetadata() local
116 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
118 ret_alloc_attrs->push_back(alloc_attr); in UpdateArgAndRetvalMetadata()
|
| D | rendezvous_util.cc | 83 AllocatorAttributes alloc_attr; in RecvOutputsFromRendezvousAsync() local
85 alloc_attr = alloc_attrs[i]; in RecvOutputsFromRendezvousAsync()
88 alloc_attr); in RecvOutputsFromRendezvousAsync()
|
| D | executor.cc | 688 AllocatorAttributes* alloc_attr) { in ExtractScopedAllocatorAttr() argument
692 CHECK_EQ(alloc_attr->scope_id, 0); in ExtractScopedAllocatorAttr()
693 alloc_attr->scope_id = sc_attr[i + 1]; in ExtractScopedAllocatorAttr()
868 alloc_attr(other.alloc_attr), in Entry()
886 alloc_attr = other.alloc_attr; in operator =()
902 alloc_attr = other.alloc_attr; in operator =()
931 AllocatorAttributes alloc_attr; member
1899 (*input_alloc_attrs)[i] = entry->alloc_attr; in PrepareInputs()
2026 out->alloc_attr = ctx->output_alloc_attr(i); in ProcessOutputs()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | host_constant_op.cc | 30 AllocatorAttributes alloc_attr; in _HostConstantOp() local
31 alloc_attr.set_on_host(true); in _HostConstantOp()
34 ctx, ctx->device()->MakeTensorFromProto(*proto, alloc_attr, &tensor_)); in _HostConstantOp()
|
| D | dynamic_partition_op_gpu.cu.cc | 245 AllocatorAttributes alloc_attr; in ComputeAsync() local
246 alloc_attr.set_on_host(true); in ComputeAsync()
250 &partition_count, alloc_attr), in ComputeAsync()
281 AllocatorAttributes alloc_attr; in ComputeAsync() local
282 alloc_attr.set_on_host(true); in ComputeAsync()
283 alloc_attr.set_gpu_compatible(true); in ComputeAsync()
287 &cpu_tensor, alloc_attr), in ComputeAsync()
|
| D | scale_and_translate_op.cc | 65 AllocatorAttributes alloc_attr; in ComputeSpansCore() local
66 alloc_attr.set_on_host(true); in ComputeSpansCore()
69 &spans->starts, alloc_attr)); in ComputeSpansCore()
74 &spans->weights, alloc_attr)); in ComputeSpansCore()
157 AllocatorAttributes alloc_attr; in ComputeGradSpansCore() local
158 alloc_attr.set_on_host(true); in ComputeGradSpansCore()
161 &grad_spans->starts, alloc_attr)); in ComputeGradSpansCore()
166 &grad_spans->weights, alloc_attr)); in ComputeGradSpansCore()
|
| D | cuda_solvers.h | 346 AllocatorAttributes alloc_attr;
350 alloc_attr.set_on_host(true);
351 alloc_attr.set_gpu_compatible(true);
354 &scratch_tensor_, alloc_attr));
|
| D | reduction_ops_common.h | 173 const AllocatorAttributes alloc_attr = ctx->output_alloc_attr(0);
180 helper.out_reshape(), &tmp_out, alloc_attr));
224 alloc_attr));
|
| D | stack.cc | 194 AllocatorAttributes alloc_attr; in Compute() local
195 alloc_attr.set_on_host(true); in Compute()
198 &stack->handle_, alloc_attr)); in Compute()
|
| D | crop_and_resize_op.cc | 786 AllocatorAttributes alloc_attr; in RunIfBoxIndexIsValid() local
787 alloc_attr.set_on_host(true); in RunIfBoxIndexIsValid()
788 alloc_attr.set_gpu_compatible(true); in RunIfBoxIndexIsValid()
792 &isvalid_host_tensor, alloc_attr), in RunIfBoxIndexIsValid()
|
| D | tensor_array_ops.cc | 111 AllocatorAttributes alloc_attr; in Compute() local
112 alloc_attr.set_on_host(true); in Compute()
115 &tensor_array_output_handle, alloc_attr)); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 775 AllocatorAttributes alloc_attr; in GetDeviceAndAllocator() local
777 dev_allocator = devices[0]->GetAllocator(alloc_attr); in GetDeviceAndAllocator()
|