Searched refs:MathGradTest (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | math_grad_test.cc | 39 class MathGradTest : public ::testing::Test { class
438 TEST_F(MathGradTest, Error_Reporting) { in TEST_F() argument
446 TEST_F(MathGradTest, Abs) { in TEST_F() argument
456 TEST_F(MathGradTest, Neg) { in TEST_F() argument
466 TEST_F(MathGradTest, Reciprocal) { in TEST_F() argument
476 TEST_F(MathGradTest, Square) { in TEST_F() argument
486 TEST_F(MathGradTest, Sqrt) { in TEST_F() argument
496 TEST_F(MathGradTest, Rsqrt) { in TEST_F() argument
506 TEST_F(MathGradTest, Exp) { in TEST_F() argument
516 TEST_F(MathGradTest, Expm1) { in TEST_F() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | math_grad_test.cc | 601 class MathGradTest : public ::testing::Test { class
603 MathGradTest() : root_(Scope::NewRootScope().WithDevice("/cpu:0")) {} in MathGradTest() function in tensorflow::__anon9fed5aaa0111::MathGradTest
676 TEST_F(MathGradTest, MatMulGrad_NoTranspose) { in TEST_F() argument
680 TEST_F(MathGradTest, MatMulComplexGrad_NoTranspose) { in TEST_F() argument
684 TEST_F(MathGradTest, MatMulGrad_TransposeX) { in TEST_F() argument
688 TEST_F(MathGradTest, MatMulComplexGrad_TransposeX) { in TEST_F() argument
692 TEST_F(MathGradTest, MatMulGrad_TransposeY) { in TEST_F() argument
696 TEST_F(MathGradTest, MatMulComplexGrad_TransposeY) { in TEST_F() argument
700 TEST_F(MathGradTest, MatMulGrad_TransposeX_TransposeY) { in TEST_F() argument
704 TEST_F(MathGradTest, MatMulComplexGrad_TransposeX_TransposeY) { in TEST_F() argument
[all …]
|