Home
last modified time | relevance | path

Searched refs:iterator_resource (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py83 def __init__(self, iterator_resource, initializer, output_types, argument
103 self._iterator_resource = iterator_resource
208 iterator_resource = gen_dataset_ops.iterator_v2(
214 return Iterator(iterator_resource, None, output_types, output_shapes,
281 iterator_resource = gen_dataset_ops.iterator_from_string_handle_v2(
285 return Iterator(iterator_resource, None, output_types, output_shapes,
892 iterator_resource, argument
896 iterator_resource, external_state_policy=external_state_policy.value)
902 device=iterator_resource.device)
904 super(_IteratorSaveable, self).__init__(iterator_resource, specs, name)
Ddataset_ops.py2490 iterator_resource = gen_dataset_ops.iterator_v2(
2495 iterator_resource)
2498 return iterator_ops.Iterator(iterator_resource, initializer,
4896 iterator_resource = op.inputs[0]
4898 op for op in iterator_resource.consumers() if op.type == "MakeIterator"
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc371 IteratorResource* iterator_resource) { in InitializeFromIterator() argument
373 TF_RETURN_IF_ERROR(iterator_resource->Save(serialization_ctx, &writer)); in InitializeFromIterator()
601 IteratorResource* iterator_resource; in DoCompute() local
603 LookupResource(ctx, HandleFromInput(ctx, 1), &iterator_resource)); in DoCompute()
604 core::ScopedUnref unref_iterator(iterator_resource); in DoCompute()
605 return iterator_resource->SetIteratorFromDataset(ctx, dataset); in DoCompute()
1054 IteratorResource* iterator_resource; in Compute() local
1056 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator_resource)); in Compute()
1057 iterator_resource->Unref(); in Compute()
1099 IteratorResource* iterator_resource; in Compute() local
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Drange_dataset_serialization_test.py45 def _save_op(self, iterator_resource): argument
47 iterator_resource)
53 def _restore_op(self, iterator_resource): argument
57 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dprefetching_ops.py175 iterator_resource = gen_dataset_ops.iterator_from_string_handle_v2(
180 iterator_resource, ignore_lookup_error=True)]):
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.-iterator.pbtxt28 …argspec: "args=[\'self\', \'iterator_resource\', \'initializer\', \'output_types\', \'output_shape…
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py60 def _save_op(self, iterator_resource): argument
62 iterator_resource)
68 def _restore_op(self, iterator_resource): argument
71 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,