Searched refs:set_gpu_compatible (Results 1 – 25 of 30) sorted by relevance
12
55 aa.set_gpu_compatible(gpu_compatible); in TEST()70 b.set_gpu_compatible(true); in TEST()82 a.set_gpu_compatible(true); in TEST()
364 void set_gpu_compatible(bool v) { value |= (static_cast<int>(v) << 2); } in set_gpu_compatible() function
47 attr.set_gpu_compatible(true); in Compute()106 attr.set_gpu_compatible(true); in Compute()
38 alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
58 attr.set_gpu_compatible(true); in EnsureSparseVariableAccess()219 attr.set_gpu_compatible(true); in PrepareToUpdateVariable()
58 attr.set_gpu_compatible(true); in Init()
59 attr.set_gpu_compatible(true); in Compute()
46 pinned_allocator.set_gpu_compatible(true); in Compute()
85 attr.set_gpu_compatible(true); in Compute()
89 attr.set_gpu_compatible(true); \
160 attr.set_gpu_compatible(true); in ComputeAsync()
253 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
351 alloc_attr.set_gpu_compatible(true);
283 alloc_attr.set_gpu_compatible(true); in ComputeAsync()
130 cpu_attr.set_gpu_compatible(true); in RecvFromPeer()
269 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
101 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
268 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
119 alloc_attrs.set_gpu_compatible(true); in GrpcRecvTensorAsync()
376 alloc_attrs.set_gpu_compatible(true); in TransportOptionsFromTensor()459 alloc_attrs.set_gpu_compatible(true); in TensorFromTransportOptions()
536 attr.set_gpu_compatible(true); in EnsureOutputAllocated()581 attr.set_gpu_compatible(true); in ProcessResult()694 attr.set_gpu_compatible(true); in ReadBatchResult()
569 attr.set_gpu_compatible(true); in GetBatch()957 attr.set_gpu_compatible(true); in AllocateOutput()
497 alloc_attrs.set_gpu_compatible(true); in GrpcRecvTensorAsync()591 cpu_attr.set_gpu_compatible(true); in RecvBufAsync()
269 allocator_attributes.set_gpu_compatible(true); in TEST_F()
476 host_alloc_attrs.set_gpu_compatible(true); in Compute()