Searched refs:x_fn (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() argument
104 x_data_flat(i) = x_fn(i); in TestCWiseGrad()
227 auto x_fn = [this](const int i) { return RV({-1, 0, 1}); }; in TEST_F() local
228 TestCWiseGrad<float, float>(ABS, x_fn); in TEST_F()
232 auto x_fn = [this](const int i) { return RV({-1, 0, 1}); }; in TEST_F() local
233 TestCWiseGrad<float, float>(NEG, x_fn); in TEST_F()
237 auto x_fn = [this](const int i) { return RV({-1, 1, -2, 2, -3, 3, -4, 4}); }; in TEST_F() local
238 TestCWiseGrad<float, float>(INV, x_fn); in TEST_F()
242 auto x_fn = [this](const int i) { return CRV({{-1, 0}, {1, 0}, {2, -1}}); }; in TEST_F() local
243 TestCWiseGrad<complex64, complex64>(INV, x_fn); in TEST_F()
[all …]
|