}

// Deterministic test value for UnsortedSegmentMin/Max, since the numerical
// gradient can be wrong if the compared inputs are nearly the same (which can
// happen with random inputs).
constexpr float kUnsortedSegmentMinMaxTestValue[] = {
    0.5f,  0.7f, 0.2f, 1.0f, 1.5f, 10.5f, -0.7f, 1.2f,
    -1.0f, 2.5f, 4.2f, 3.7f, 1.2f, -5.0f, -1.5f};

TEST_F(NaryGradTest, UnsortedSegmentMaxGrad) {
  TensorShape shape({3, 1, 5});