Home
last modified time | relevance | path

Searched refs:gpu_compatible (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc51 for (bool gpu_compatible : {false, true}) { in TEST()
55 aa.set_gpu_compatible(gpu_compatible); in TEST()
58 EXPECT_EQ(gpu_compatible, aa.gpu_compatible()); in TEST()
Dallocator.h365 bool gpu_compatible() const { return value & (0x1 << 2); } in gpu_compatible() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc46 if (attr.gpu_compatible() || force_gpu_compatible_) { in GetAllocator()
96 if (attr.gpu_compatible() || force_gpu_compatible_) { in GetAllocator()
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc101 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
102 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc269 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
270 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto86 // Force all tensors to be gpu_compatible. On a GPU-enabled TensorFlow,