| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_solvers.h | 248 bool on_host);
252 bool on_host);
565 ScratchSpace(OpKernelContext* context, int64_t size, bool on_host)
566 : ScratchSpace(context, TensorShape({size}), "", on_host) {}
569 const std::string& debug_info, bool on_host)
570 : ScratchSpace(context, TensorShape({size}), debug_info, on_host) {}
573 const std::string& debug_info, bool on_host)
574 : context_(context), debug_info_(debug_info), on_host_(on_host) {
576 if (on_host) {
608 bool on_host() const { return on_host_; }
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 46 (send_args.alloc_attrs.on_host() || parsed.src.type == "CPU"); in SameWorkerRecvDone()
48 (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU"); in SameWorkerRecvDone()
52 send_args.alloc_attrs.on_host() && !(parsed.src.type == "CPU"); in SameWorkerRecvDone()
54 recv_args.alloc_attrs.on_host() && !(parsed.dst.type == "CPU"); in SameWorkerRecvDone()
|
| D | single_threaded_executor.cc | 202 bool on_host = in Initialize() local
204 kernel_state.output_alloc_attr.set_on_host(on_host); in Initialize()
229 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize() local
230 if (on_host) { in Initialize()
232 h.set_on_host(on_host); in Initialize()
|
| 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 | graph_view.cc | 449 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs() local
450 if (on_host) { in SetAllocAttrs()
452 h.set_on_host(on_host); in SetAllocAttrs()
|
| D | copy_tensor.cc | 214 src_alloc_attr.on_host() ? DEVICE_CPU : src->attributes().device_type()); in ViaDMA()
216 dst_alloc_attr.on_host() ? DEVICE_CPU : dst->attributes().device_type()); in ViaDMA()
|
| D | partitioning_utils_test.cc | 273 ASSERT_EQ(expected[i], actual[i].on_host()) << " at index " << i; in CheckAlloc()
|
| /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 | 1139 alloc_attrs.on_host = 0; in TEST_F()
1141 alloc_attrs.on_host = 1; in TEST_F()
1176 alloc_attrs.on_host = 0; in TEST_F()
1178 alloc_attrs.on_host = 1; in TEST_F()
1209 alloc_attrs.on_host = 0; in TEST_F()
1211 alloc_attrs.on_host = 1; in TEST_F()
1250 alloc_attrs.on_host = 0; in TEST_F()
1252 alloc_attrs.on_host = 1; in TEST_F()
|
| /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/framework/ |
| D | allocator_test.cc | 56 for (bool on_host : {false, true}) { in TEST()
60 aa.set_on_host(on_host); in TEST()
63 EXPECT_EQ(on_host, aa.on_host()); in TEST()
|
| D | allocator.cc | 61 return strings::StrCat("AllocatorAttributes(on_host=", on_host(), in DebugString()
|
| D | allocator.h | 315 bool on_host() const { return value & 0x1; } in on_host() function
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 530 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync() local
533 if (src_dev->tensorflow_accelerator_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
667 const bool on_host = in RecvBufAsync() local
669 hook->prod_attr.on_host(); in RecvBufAsync()
670 if ((!on_host) && (num_bytes > 0)) { in RecvBufAsync()
|
| /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 | function_ops.cc | 372 bool on_host = DataTypeAlwaysOnHost(dtype); in ComputeAsync() local
379 on_host = true; in ComputeAsync()
387 ret_alloc_attrs.set_on_host(on_host); in ComputeAsync()
|
| D | stack.cc | 243 if (swap_memory_ && !alloc_attrs.on_host() && in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device.cc | 245 if (attr.on_host()) { in GetAllocator()
323 if (alloc_attrs.on_host()) { in MaybeCopyTensorToPluggableDevice()
|
| /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 | 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 | 398 TFE_TensorHandle* on_host = in TEST() local
401 TF_Tensor* resolved = TFE_TensorHandleResolve(on_host, status); in TEST()
408 TFE_DeleteTensorHandle(on_host); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 267 if (attr.on_host()) { in GetAllocatorLocked()
515 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/compiler/tf2xla/kernels/ |
| D | light_outside_compilation.cc | 346 if (attr.on_host()) { in GetAllocator()
|