Searched refs:set_gpu_compatible (Results 1 – 25 of 29) 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()
311 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()
59 attr.set_gpu_compatible(true); in EnsureSparseVariableAccess()218 attr.set_gpu_compatible(true); in PrepareToUpdateVariable()
59 attr.set_gpu_compatible(true); in Init()
59 attr.set_gpu_compatible(true); in Compute()
56 pinned_allocator.set_gpu_compatible(true); in Compute()
84 attr.set_gpu_compatible(true); in Compute()
89 attr.set_gpu_compatible(true); \
116 alloc_attr.set_gpu_compatible(true);
253 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
237 attr.set_gpu_compatible(true); in ComputeAsync()
391 alloc_attr.set_gpu_compatible(true);
294 alloc_attr.set_gpu_compatible(true); in ComputeAsync()
135 attr.set_gpu_compatible(true); in Compute()
295 alloc_attr.set_gpu_compatible(true); in NmsGpu()
71 host_alloc_attrs.set_gpu_compatible(true); in RunComposedOp()
131 cpu_attr.set_gpu_compatible(true); in RecvFromPeer()
280 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
79 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
220 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
474 attr.set_gpu_compatible(true); in EnsureOutputAllocated()519 attr.set_gpu_compatible(true); in ProcessResult()632 attr.set_gpu_compatible(true); in ReadBatchResult()
531 alloc_attrs.set_gpu_compatible(true); in GrpcRecvTensorAsync()670 cpu_attr.set_gpu_compatible(true); in RecvBufAsync()
298 allocator_attributes.set_gpu_compatible(true); in TEST_F()
519 host_alloc_attrs.set_gpu_compatible(true); in Compute()