| D | math_grad_test.cc | 280 Tensor BatchMatMulV2(const Tensor& x, bool ax, const Tensor& y, bool ay) { in BatchMatMulV2() function in tensorflow::__anona51524880111::MathGradTest
1238 test::ExpectClose(dx, BatchMatMulV2(dz, false, y, true)); in TEST_F()
1239 test::ExpectClose(dy, BatchMatMulV2(x, true, dz, false)); in TEST_F()
1250 test::ExpectClose(dx, BatchMatMulV2(dz, false, y, false)); in TEST_F()
1251 test::ExpectClose(dy, BatchMatMulV2(dz, true, x, false)); in TEST_F()
1262 test::ExpectClose(dx, BatchMatMulV2(y, false, dz, true)); in TEST_F()
1263 test::ExpectClose(dy, BatchMatMulV2(x, false, dz, false)); in TEST_F()
1274 test::ExpectClose(dx, BatchMatMulV2(y, true, dz, true)); in TEST_F()
1275 test::ExpectClose(dy, BatchMatMulV2(dz, true, x, true)); in TEST_F()
1292 CHECK(ans_dx.CopyFrom(ReduceSum(BatchMatMulV2(dz, false, y, true), {0}), in TEST_F()
[all …]
|