Python _EagerTensorCache Exemples

Exemple #1

    def testCacheSkipsTensorsTooLarge(self):
        cache = context._EagerTensorCache(max_items=100, max_tensor_size=3)
        cache.put('1', array_ops.zeros((2, 2)))
        self.assertEqual(cache.get('1'), None)

        cache.put('2', array_ops.zeros((2)))
        self.assertNotEqual(cache.get('2'), None)

Exemple #2

Fichier : core_test.py Projet : kylin9872/tensorflow

  def testCacheSkipsTensorsTooLarge(self):
    cache = context._EagerTensorCache(max_items=100, max_tensor_size=3)
    cache.put('1', array_ops.zeros((2, 2)))
    self.assertEqual(cache.get('1'), None)

    cache.put('2', array_ops.zeros((2)))
    self.assertNotEqual(cache.get('2'), None)