D | math_grad_test.cc | 215 complex64 CRV(const std::vector<complex64>& v) { in CRV() function in tensorflow::__anonf4b1ef9d0111::CWiseUnaryGradTest 242 auto x_fn = [this](const int i) { return CRV({{-1, 0}, {1, 0}, {2, -1}}); }; in TEST_F() 252 auto x_fn = [this](const int i) { return CRV({{-1, 0}, {1, 0}, {2, -1}}); }; in TEST_F() 263 return CRV({{-1.0f, 0.5f}, {1.0f, 0.5f}, {2, -1}}); in TEST_F() 275 return CRV({{-1.0f, 0.5f}, {1.0f, 0.5f}, {2, -1}}); in TEST_F() 288 auto x_fn = [this](const int i) { return CRV({{-1, 0}, {1, 0}, {2, -1}}); }; in TEST_F() 299 return CRV({{-1, 0}, {1, 0}, {1.5, -1.5}}); in TEST_F() 311 return CRV({{-1, 0.5f}, {1, 0.5f}, {2, -1}}); in TEST_F() 323 return CRV({{0, 0}, {1e-6, 0}, {2, -1}, {1, 2}, {3, 4}}); in TEST_F() 335 return CRV({{0.5, 0.25}, {0.25, 0.5}, {1.5, -1}, {1, 1.5}}); in TEST_F() [all …]
|