| H A D | math_grad_test.cc | 94 void TestCWiseGrad(UnaryOpType op_type, const std::function<X_T(int)>& x_fn) { argument
102 x_data_flat(i) = x_fn(i);
225 auto x_fn = [this](const int i) { return RV({-1, 0, 1}); }; local
226 TestCWiseGrad<float, float>(ABS, x_fn);
230 auto x_fn = [this](const int i) { return RV({-1, 0, 1}); }; local
231 TestCWiseGrad<float, float>(NEG, x_fn);
235 auto x_fn = [this](const int i) { return RV({-1, 1, -2, 2, -3, 3, -4, 4}); }; local
236 TestCWiseGrad<float, float>(INV, x_fn);
240 auto x_fn = [this](const int i) { return CRV({{-1, 0}, {1, 0}, {2, -1}}); }; local
241 TestCWiseGrad<complex64, complex64>(INV, x_fn);
245 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
250 auto x_fn = [this](const int i) { return CRV({{-1, 0}, {1, 0}, {2, -1}}); }; local
255 auto x_fn = [this](const int i) { return RV({0.5, 1, 2, 3, 4, 5, 6, 7}); }; local
260 auto x_fn = [this](const int i) { local
267 auto x_fn = [this](const int i) { return RV({1, 2, 3, 4, 5, 6, 7, 8}); }; local
272 auto x_fn = [this](const int i) { local
279 auto x_fn = [this](const int i) { local
286 auto x_fn = [this](const int i) { return CRV({{-1, 0}, {1, 0}, {2, -1}}); }; local
291 auto x_fn = [this](const int i) { return RV({0, -1, 1e-6, 1, -1.5, 1.5}); }; local
296 auto x_fn = [this](const int i) { local
303 auto x_fn = [this](const int i) { return RV({0.5, 1, 2, 3, 4}); }; local
308 auto x_fn = [this](const int i) { local
315 auto x_fn = [this](const int i) { return RV({0, 1e-6, 1, 2, 3, 4, 100}); }; local
320 auto x_fn = [this](const int i) { local
327 auto x_fn = [this](const int i) { return RV({0.5, -0.5, 1, -1, 1.5, -1.5}); }; local
332 auto x_fn = [this](const int i) { local
339 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
344 auto x_fn = [this](const int i) { local
351 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
356 auto x_fn = [this](const int i) { local
363 auto x_fn = [this](const int i) { return RV({0.5, 1, -1, -1.5, 1.5}); }; local
368 auto x_fn = [this](const int i) { local
375 auto x_fn = [this](const int i) { return RV({1.5, 2, 2.5}); }; local
380 auto x_fn = [this](const int i) { local
387 auto x_fn = [this](const int i) { return RV({0, -0.5, 0.5, -0.1, 0.1}); }; local
392 auto x_fn = [this](const int i) { local
399 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
404 auto x_fn = [this](const int i) { local
411 auto x_fn = [this](const int i) { return RV({-1, 1, -2, 2, -3, 3}); }; local
416 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
421 auto x_fn = [this](const int i) { local
428 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
433 auto x_fn = [this](const int i) { local
440 auto x_fn = [this](const int i) { return RV({0, 0.25, -0.25, -0.5, 0.5}); }; local
445 auto x_fn = [this](const int i) { local
456 auto x_fn = [this](const int i) { return RV({0, -0.5, 0.5, -0.75, 0.75}); }; local
461 auto x_fn = [this](const int i) { local
472 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
477 auto x_fn = [this](const int i) { local
488 auto x_fn = [this](const int i) { return RV({0, -1, 1, -2, 2, -3, 3}); }; local
493 auto x_fn = [this](const int i) { local
504 auto x_fn = [this](const int i) { local
511 auto x_fn = [this](const int i) { local
518 auto x_fn = [this](const int i) { local
525 auto x_fn = [this](const int i) { return RV({1, -1, 2, -2, 3, -3}); }; local
530 auto x_fn = [this](const int i) { local
537 auto x_fn = [this](const int i) { local
544 auto x_fn = [this](const int i) { local
555 auto x_fn = [this](const int i) { local
562 auto x_fn = [this](const int i) { local
[all...] |