| 964 | public ::testing::WithParamInterface<std::tuple<bool, bool, int>> {}; |
| 965 | |
| 966 | TEST_P(CumsumGradTest, CumsumGrad) { |
| 967 | int axis = std::get<2>(GetParam()); |
| 968 | |
| 969 | TensorShape shape({2, 3, 2}); |
| 970 | auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape)); |
| 971 | Cumsum::Attrs attrs; |
| 972 | attrs.exclusive_ = std::get<0>(GetParam()); |
| 973 | attrs.reverse_ = std::get<1>(GetParam()); |
| 974 | auto y = Cumsum(scope_, x, axis, attrs); |
| 975 | RunTest({x}, {shape}, {y}, {shape}); |
| 976 | } |
| 977 | |
| 978 | INSTANTIATE_TEST_SUITE_P(CumsumGrad, CumsumGradTest, |
| 979 | ::testing::Combine(::testing::Bool(), |
nothing calls this directly
no test coverage detected