Searched refs:TestCWiseGrad (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| H A D | math_grad_test.cc | 94 void TestCWiseGrad(UnaryOpType op_type, const std::function<X_T(int)>& x_fn) { function in class:tensorflow::__anon25356::CWiseUnaryGradTest
226 TestCWiseGrad<float, float>(ABS, x_fn);
231 TestCWiseGrad<float, float>(NEG, x_fn);
236 TestCWiseGrad<float, float>(INV, x_fn);
241 TestCWiseGrad<complex64, complex64>(INV, x_fn);
246 TestCWiseGrad<float, float>(SQUARE, x_fn);
251 TestCWiseGrad<complex64, complex64>(SQUARE, x_fn);
256 TestCWiseGrad<float, float>(SQRT, x_fn);
263 TestCWiseGrad<complex64, complex64>(SQRT, x_fn);
268 TestCWiseGrad<floa
[all...] |
Completed in 215 milliseconds