Searched refs:TestCWiseGrad (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad_test.cc | 96 void TestCWiseGrad(UnaryOpType op_type, const std::function<X_T(int)>& x_fn) { in TestCWiseGrad() function in tensorflow::__anonf4b1ef9d0111::CWiseUnaryGradTest
228 TestCWiseGrad<float, float>(ABS, x_fn); in TEST_F()
233 TestCWiseGrad<float, float>(NEG, x_fn); in TEST_F()
238 TestCWiseGrad<float, float>(INV, x_fn); in TEST_F()
243 TestCWiseGrad<complex64, complex64>(INV, x_fn); in TEST_F()
248 TestCWiseGrad<float, float>(SQUARE, x_fn); in TEST_F()
253 TestCWiseGrad<complex64, complex64>(SQUARE, x_fn); in TEST_F()
258 TestCWiseGrad<float, float>(SQRT, x_fn); in TEST_F()
265 TestCWiseGrad<complex64, complex64>(SQRT, x_fn); in TEST_F()
270 TestCWiseGrad<float, float>(RSQRT, x_fn); in TEST_F()
[all …]
|