Searched refs:gpu_compatible (Results 1 – 8 of 8) sorted by relevance
52 for (bool gpu_compatible : {false, true}) { in TEST()56 aa.set_gpu_compatible(gpu_compatible); in TEST()59 EXPECT_EQ(gpu_compatible, aa.gpu_compatible()); in TEST()
65 " gpu_compatible=", gpu_compatible(), ")"); in DebugString()
371 bool gpu_compatible() const { return value & (0x1 << 2); } in gpu_compatible() function
47 if (attr.gpu_compatible() || force_gpu_compatible_) { in GetAllocator()97 if (attr.gpu_compatible() || force_gpu_compatible_) { in GetAllocator()
113 name, value, value_fn, gpu_compatible = y117 gpu_compatible=gpu_compatible)366 gpu_compatible): argument367 if not gpu_compatible and test.is_gpu_available():
80 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()81 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
265 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()266 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
88 // Force all tensors to be gpu_compatible. On a GPU-enabled TensorFlow,