Python map_fn Exemples

Exemple #1

Fichier : map_test.py Projet : yz3052/tensorflow-upstream

  def testShortCircuit(self, structure, map_fn, num_parallel_calls):
    dataset = self.structuredDataset(structure).repeat().map(
        map_fn, num_parallel_calls=num_parallel_calls)
    get_next = self.getNext(dataset)

    if isinstance(structure, tuple):
      expected = map_fn(*self.evaluate(self.structuredElement(structure)))
    else:
      expected = map_fn(self.evaluate(self.structuredElement(structure)))
    self.assertEqual(expected, self.evaluate(get_next()))

Exemple #2

Fichier : map_test.py Projet : perfmjs/tensorflow

  def testShortCircuit(self, structure, map_fn, num_parallel_calls):
    dataset = self.structuredDataset(structure).repeat().map(
        map_fn, num_parallel_calls=num_parallel_calls)
    get_next = self.getNext(dataset)

    if isinstance(structure, tuple):
      expected = map_fn(*self.evaluate(self.structuredElement(structure)))
    else:
      expected = map_fn(self.evaluate(self.structuredElement(structure)))
    self.assertEqual(expected, self.evaluate(get_next()))