| /external/tensorflow/tensorflow/core/kernels/ |
| D | rocm_solvers.h | 101 ScratchSpace(OpKernelContext* context, int64 size, bool on_host)
102 : ScratchSpace(context, TensorShape({size}), "", on_host) {}
105 bool on_host)
106 : ScratchSpace(context, TensorShape({size}), debug_info, on_host) {}
109 const string& debug_info, bool on_host)
110 : context_(context), debug_info_(debug_info), on_host_(on_host) {
112 if (on_host) {
144 bool on_host() const { return on_host_; }
|
| D | cuda_solvers.h | 171 const string& debug_info, bool on_host);
174 bool on_host);
376 ScratchSpace(OpKernelContext* context, int64 size, bool on_host)
377 : ScratchSpace(context, TensorShape({size}), "", on_host) {}
380 bool on_host)
381 : ScratchSpace(context, TensorShape({size}), debug_info, on_host) {}
384 const string& debug_info, bool on_host)
385 : context_(context), debug_info_(debug_info), on_host_(on_host) {
387 if (on_host) {
419 bool on_host() const { return on_host_; }
[all …]
|
| D | ops_testutil.cc | 60 const bool on_host = in SetOutputAttrs() local
62 attr.set_on_host(on_host); in SetOutputAttrs()
|
| D | stack.cc | 242 if (swap_memory_ && !alloc_attrs.on_host() && in ComputeAsync()
|
| D | debug_ops.h | 89 !context->input_alloc_attr(0).on_host(); in Compute()
107 !context->input_alloc_attr(0).on_host(); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator_test.cc | 50 for (bool on_host : {false, true}) { in TEST()
54 aa.set_on_host(on_host); in TEST()
57 EXPECT_EQ(on_host, aa.on_host()); in TEST()
|
| D | allocator.cc | 62 return strings::StrCat("AllocatorAttributes(on_host=", on_host(), in DebugString()
|
| D | allocator.h | 308 bool on_host() const { return value & 0x1; } in on_host() function
|
| /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/sycl/ |
| D | sycl_device.cc | 39 if (attr.on_host()) in GetAllocator()
58 if (alloc_attrs.on_host()) { in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/c/experimental/ |
| D | rendezvous.cc | 65 alloc_attrs->on_host = args.alloc_attrs.on_host(); in RecvFromRemoteAsync()
|
| D | rendezvous_internal.h | 39 bool on_host; member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 525 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync() local
528 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
662 const bool on_host = in RecvBufAsync() local
664 hook->prod_attr.on_host(); in RecvBufAsync()
665 if ((!on_host) && (num_bytes > 0)) { in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 124 src_attr.on_host() ? DEVICE_CPU : src_dev->attributes().device_type()); in MemCpyAsync()
126 dst_attr.on_host() ? DEVICE_CPU : dst_dev->attributes().device_type()); in MemCpyAsync()
|
| D | rendezvous_mgr.cc | 44 (send_args.alloc_attrs.on_host() || parsed.src.type == "CPU"); in SameWorkerRecvDone()
46 (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU"); in SameWorkerRecvDone()
|
| D | copy_tensor.cc | 211 src_alloc_attr.on_host() ? DEVICE_CPU : src->attributes().device_type()); in ViaDMA()
213 dst_alloc_attr.on_host() ? DEVICE_CPU : dst->attributes().device_type()); in ViaDMA()
|
| D | partitioning_utils_test.cc | 166 ASSERT_EQ(expected[i], actual[i].on_host()) << " at index " << i; in CheckAlloc()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 170 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize() local
171 if (on_host) { in Initialize()
173 h.set_on_host(on_host); in Initialize()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 46 if (attr.on_host()) { in GetAllocator()
|
| D | gpu_event_mgr_test.cc | 414 const bool on_host = in SetOutputAttrs() local
416 attr.set_on_host(on_host); in SetOutputAttrs()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 241 (send_args.alloc_attrs.on_host() || parsed.src.type == "CPU"); in SameWorkerRecvDone()
243 (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU"); in SameWorkerRecvDone()
|
| D | tensor_coding.cc | 47 if (alloc_attrs_.on_host() || da.device_type() == "CPU") { in InitAlloc()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental_test.cc | 529 TFE_TensorHandle* on_host = in TEST() local
532 TF_Tensor* resolved = TFE_TensorHandleResolve(on_host, status); in TEST()
539 TFE_DeleteTensorHandle(on_host); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 257 if (attr.on_host()) { in GetAllocatorLocked()
485 if (alloc_attrs.on_host()) { in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 1101 const bool on_host = IsDstInputOnHost(edge, g_info); in Partition() local
1102 DupRecvKey key{src->id(), edge->src_output(), dst_graph, on_host}; in Partition()
|