def _assert_sparse_tensor_equals(self, sp1, sp2):
    self.assertAllEqual(sp1.indices, sp2.indices)
    self.assertAllEqual(sp1.values, sp2.values)
    self.assertAllEqual(sp1.dense_shape, sp2.dense_shape)