Searched refs:set_gpu_compatible (Results 1 – 25 of 30) sorted by relevance
12
56 aa.set_gpu_compatible(gpu_compatible); in TEST()71 b.set_gpu_compatible(true); in TEST()83 a.set_gpu_compatible(true); in TEST()
370 void set_gpu_compatible(bool v) { value |= (static_cast<int>(v) << 2); } in set_gpu_compatible() function
38 alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
60 attr.set_gpu_compatible(true); in Init()
59 attr.set_gpu_compatible(true); in EnsureSparseVariableAccess()218 attr.set_gpu_compatible(true); in PrepareToUpdateVariable()
46 pinned_allocator.set_gpu_compatible(true); in Compute()
59 attr.set_gpu_compatible(true); in Compute()
84 attr.set_gpu_compatible(true); in Compute()
89 attr.set_gpu_compatible(true); \
254 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
240 attr.set_gpu_compatible(true); in ComputeAsync()
115 attr.set_gpu_compatible(true); in Compute()
290 alloc_attr.set_gpu_compatible(true); in ComputeAsync()
77 host_alloc_attrs.set_gpu_compatible(true); in RunComposedOp()
117 alloc_attr.set_gpu_compatible(true);
394 alloc_attr.set_gpu_compatible(true);
361 attr->set_gpu_compatible(true); in InferAllocAttr()388 attr->set_gpu_compatible(true); in InferAllocAttr()
80 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
221 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
134 cpu_attr.set_gpu_compatible(true); in RecvFromPeer()
265 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
531 alloc_attrs.set_gpu_compatible(true); in GrpcRecvTensorAsync()670 cpu_attr.set_gpu_compatible(true); in RecvBufAsync()
734 attr.set_gpu_compatible(true); in ReadBatch()830 attr.set_gpu_compatible(true); in ProcessBatch()
396 allocator_attributes.set_gpu_compatible(true); in TEST_F()
422 host_alloc_attrs.set_gpu_compatible(true); in Compute()