Esempi in Python per experimental_identity_indexed_dataset

Esempio n. 1

File: indexed_dataset_ops_test.py Progetto: GiveMeLuna/tensorflow

    def testLowLevelIndexedDatasetOps(self):
        identity = ged_ops.experimental_identity_indexed_dataset(
            ops.convert_to_tensor(16, dtype=dtypes.uint64))
        handle = ged_ops.experimental_materialized_index_dataset_handle(
            container="",
            shared_name="",
            output_types=[dtypes.uint64],
            output_shapes=[[]])
        materialize = ged_ops.experimental_indexed_dataset_materialize(
            identity, handle)
        get_op = ged_ops.experimental_indexed_dataset_get(
            handle, 3, output_types=[dtypes.uint64], output_shapes=[[]])

        self.evaluate(materialize)
        self.assertEqual([3], self.evaluate(get_op))

Esempio n. 2

File: indexed_dataset_ops_test.py Progetto: Wajih-O/tensorflow

  def testLowLevelIndexedDatasetOps(self):
    identity = ged_ops.experimental_identity_indexed_dataset(
        ops.convert_to_tensor(16, dtype=dtypes.uint64))
    handle = ged_ops.experimental_materialized_index_dataset_handle(
        container="",
        shared_name="",
        output_types=[dtypes.uint64],
        output_shapes=[[]])
    materialize = ged_ops.experimental_indexed_dataset_materialize(
        identity, handle)
    get_op = ged_ops.experimental_indexed_dataset_get(
        handle, 3, output_types=[dtypes.uint64], output_shapes=[[]])

    self.evaluate(materialize)
    self.assertEqual([3], self.evaluate(get_op))

Esempio n. 3

  def testLowLevelIndexedDatasetOps(self):
    identity = ged_ops.experimental_identity_indexed_dataset(
        ops.convert_to_tensor(16, dtype=dtypes.uint64))
    handle = ged_ops.experimental_materialized_index_dataset_handle(
        container="",
        shared_name="",
        output_types=[dtypes.uint64],
        output_shapes=[[]])
    materialize = ged_ops.experimental_indexed_dataset_materialize(
        identity, handle)
    index = array_ops.placeholder(dtypes.uint64)
    get_op = ged_ops.experimental_indexed_dataset_get(
        handle, index, output_types=[dtypes.uint64], output_shapes=[[]])

    with self.cached_session() as sess:
      sess.run(materialize)
      self.assertEqual([3], sess.run(get_op, feed_dict={index: 3}))

Esempio n. 4

 def _as_variant_tensor(self):
   return ged_ops.experimental_identity_indexed_dataset(self._size)

Esempio n. 5

File: indexed_dataset_ops.py Progetto: JonathanRaiman/tensorflow

 def _as_variant_tensor(self):
   return ged_ops.experimental_identity_indexed_dataset(self._size)