HomeSort by relevance Sort by last modified time
    Searched refs:tf_gpu (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
basic_gpu_test.py 47 tf_gpu = sess.run(out)
55 self.assertAllClose(tf_cpu, tf_gpu)
217 tf_gpu = out.eval()
218 self.assertAllClose(np_ans, tf_gpu)
cwise_ops_test.py 157 tf_gpu = result.eval()
159 self.assertAllClose(np_ans, tf_gpu, rtol=1e-3, atol=1e-3)
161 self.assertAllClose(np_ans, tf_gpu)
634 tf_gpu = out.eval()
635 self.assertAllClose(np_ans, tf_gpu)
    [all...]

Completed in 72 milliseconds