Method _set_size

tensorflow/python/kernel_tests/sets_test.py:162–174

(self, sparse_data)

      ], self._set_size(sp))

  def _set_size(self, sparse_data):
    # Validate that we get the same results with or without `validate_indices`.
    ops = [
        sets.set_size(sparse_data, validate_indices=True),
        sets.set_size(sparse_data, validate_indices=False)
    ]
    for op in ops:
      self.assertEqual(None, op.get_shape().dims)
      self.assertEqual(dtypes.int32, op.dtype)
    with self.cached_session() as sess:
      results = self.evaluate(ops)
    self.assertAllEqual(results[0], results[1])
    return results[0]

  @test_util.run_deprecated_v1
  def test_set_intersection_multirow_2d(self):