D | math_grad_test.cc | 228 Tensor ReduceSum(const Tensor& x, gtl::ArraySlice<int32> axes) { in ReduceSum() function in tensorflow::__anona51524880111::MathGradTest 1292 CHECK(ans_dx.CopyFrom(ReduceSum(BatchMatMulV2(dz, false, y, true), {0}), in TEST_F() 1312 CHECK(ans_dy.CopyFrom(ReduceSum(BatchMatMulV2(x, true, dz, false), {0}), in TEST_F() 1332 CHECK(ans_dx.CopyFrom(ReduceSum(BatchMatMulV2(dz, false, y, true), {1}), in TEST_F() 1334 CHECK(ans_dy.CopyFrom(ReduceSum(BatchMatMulV2(x, true, dz, false), {0}), in TEST_F() 1355 CHECK(ans_dx.CopyFrom(ReduceSum(BatchMatMulV2(y, true, dz, true), {1}), in TEST_F() 1357 CHECK(ans_dy.CopyFrom(ReduceSum(BatchMatMulV2(dz, true, x, true), {0}), in TEST_F()
|