dict(
          ragged_reduce_op=ragged_math_ops.reduce_std,
          rt_input=[[3, 1, 4], [3, 1], [2], [2, 1]],
          axis=0,
          keepdims=False,