Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc52 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()
Dallocator.cc65 " gpu_compatible=", gpu_compatible(), ")"); in DebugString()
Dallocator.h371 bool gpu_compatible() const { return value & (0x1 << 2); } in gpu_compatible() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc47 if (attr.gpu_compatible() || force_gpu_compatible_) { in GetAllocator()
97 if (attr.gpu_compatible() || force_gpu_compatible_) { in GetAllocator()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptional_test.py113 name, value, value_fn, gpu_compatible = y
117 gpu_compatible=gpu_compatible)
366 gpu_compatible): argument
367 if not gpu_compatible and test.is_gpu_available():
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc80 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
81 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc265 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
266 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto88 // Force all tensors to be gpu_compatible. On a GPU-enabled TensorFlow,