| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_solvers.h | 102 ScratchSpace(OpKernelContext* context, int64 size, bool on_host)
103 : ScratchSpace(context, TensorShape({size}), "", on_host) {}
106 bool on_host)
107 : ScratchSpace(context, TensorShape({size}), debug_info, on_host) {}
110 const string& debug_info, bool on_host)
111 : context_(context), debug_info_(debug_info), on_host_(on_host) {
113 if (on_host) {
145 bool on_host() const { return on_host_; }
|
| D | cuda_solvers.h | 173 bool on_host);
177 bool on_host);
379 ScratchSpace(OpKernelContext* context, int64 size, bool on_host)
380 : ScratchSpace(context, TensorShape({size}), "", on_host) {}
383 const std::string& debug_info, bool on_host)
384 : ScratchSpace(context, TensorShape({size}), debug_info, on_host) {}
387 const std::string& debug_info, bool on_host)
388 : context_(context), debug_info_(debug_info), on_host_(on_host) {
390 if (on_host) {
422 bool on_host() const { return on_host_; }
[all …]
|
| /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 | 63 return strings::StrCat("AllocatorAttributes(on_host=", on_host(), in DebugString()
|
| D | allocator.h | 367 bool on_host() const { return value & 0x1; } in on_host() function
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.h | 53 TF_Bool on_host; member
57 TF_OFFSET_OF_END(TF_AllocatorAttributes, on_host)
|
| D | kernels_test.cc | 819 alloc_attrs.on_host = 0; in TEST_F()
821 alloc_attrs.on_host = 1; in TEST_F()
856 alloc_attrs.on_host = 0; in TEST_F()
858 alloc_attrs.on_host = 1; in TEST_F()
889 alloc_attrs.on_host = 0; in TEST_F()
891 alloc_attrs.on_host = 1; in TEST_F()
|
| D | kernels.cc | 531 if (attributes && attributes->on_host) { in TF_AllocateTemp()
|
| /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/kernels/data/ |
| D | single_threaded_executor.cc | 188 bool on_host = in Initialize() local
190 kernel_state.output_alloc_attr.set_on_host(on_host); in Initialize()
215 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize() local
216 if (on_host) { in Initialize()
218 h.set_on_host(on_host); in Initialize()
|
| /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 | 130 src_attr.on_host() ? DEVICE_CPU : src_dev->attributes().device_type()); in MemCpyAsync()
132 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 | graph_view.cc | 432 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs() local
433 if (on_host) { in SetAllocAttrs()
435 h.set_on_host(on_host); in SetAllocAttrs()
|
| D | copy_tensor.cc | 212 src_alloc_attr.on_host() ? DEVICE_CPU : src->attributes().device_type()); in ViaDMA()
214 dst_alloc_attr.on_host() ? DEVICE_CPU : dst->attributes().device_type()); in ViaDMA()
|
| D | partitioning_utils_test.cc | 182 ASSERT_EQ(expected[i], actual[i].on_host()) << " at index " << i; in CheckAlloc()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 61 const bool on_host = in SetOutputAttrs() local
63 attr.set_on_host(on_host); in SetOutputAttrs()
|
| D | stack.cc | 243 if (swap_memory_ && !alloc_attrs.on_host() && in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 46 if (attr.on_host()) { in GetAllocator()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 228 (send_args.alloc_attrs.on_host() || parsed.src.type == "CPU"); in SameWorkerRecvDone()
230 (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 | 395 TFE_TensorHandle* on_host = in TEST() local
398 TF_Tensor* resolved = TFE_TensorHandleResolve(on_host, status); in TEST()
405 TFE_DeleteTensorHandle(on_host); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 258 if (attr.on_host()) { in GetAllocatorLocked()
485 if (alloc_attrs.on_host()) { in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 269 const bool on_host = in SetOutputAttrs() local
271 attr.set_on_host(on_host); in SetOutputAttrs()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 1103 const bool on_host = IsDstInputOnHost(edge, g_info); in Partition() local
1104 DupRecvKey key{src->id(), edge->src_output(), dst_graph, on_host}; in Partition()
|