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

  /external/tensorflow/tensorflow/python/kernel_tests/
basic_gpu_test.py 53 tf_cpu = sess.run(out)
55 self.assertAllClose(tf_cpu, tf_gpu)
cwise_ops_test.py 103 tf_cpu = y.eval()
106 self.assertAllClose(np_ans, tf_cpu, rtol=1e-3, atol=1e-3)
108 self.assertAllClose(np_ans, tf_cpu)
537 tf_cpu = out.eval()
552 self.assertAllClose(np_ans, tf_cpu)
    [all...]

Completed in 60 milliseconds