| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 74 def __init__(self, iterator_resource, initializer, output_types, argument
94 self._iterator_resource = iterator_resource
198 iterator_resource = gen_dataset_ops.iterator_v2(
204 iterator_resource = gen_dataset_ops.iterator_v2(
210 iterator_resource = gen_dataset_ops.iterator(
216 return Iterator(iterator_resource, None, output_types, output_shapes,
285 iterator_resource = gen_dataset_ops.iterator_from_string_handle_v2(
290 iterator_resource = gen_dataset_ops.iterator_from_string_handle_v2(
295 iterator_resource = gen_dataset_ops.iterator_from_string_handle(
300 return Iterator(iterator_resource, None, output_types, output_shapes,
[all …]
|
| D | dataset_ops.py | 1516 iterator_resource = gen_dataset_ops.iterator_v2(
1519 iterator_resource = gen_dataset_ops.iterator(
1521 with ops.colocate_with(iterator_resource):
1524 iterator_resource)
1527 iterator_resource, initializer, get_legacy_output_types(dataset),
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | iterator_ops.py | 72 def __init__(self, iterator_resource): argument
73 serialized_iterator = gen_dataset_ops.serialize_iterator(iterator_resource)
76 iterator_resource.name + "-state")
78 super(_Saveable, self).__init__(iterator_resource, specs,
79 iterator_resource.name)
|
| D | prefetching_ops.py | 175 iterator_resource = gen_dataset_ops.iterator_from_string_handle_v2(
180 iterator_resource, ignore_lookup_error=True)]):
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 339 IteratorResource* iterator_resource) { in InitializeFromIterator() argument
346 TF_RETURN_IF_ERROR(iterator_resource->Save(&serialization_ctx, &writer)); in InitializeFromIterator()
555 IteratorResource* iterator_resource; in Compute() local
557 ctx, LookupResource(ctx, HandleFromInput(ctx, 1), &iterator_resource)); in Compute()
558 core::ScopedUnref unref(iterator_resource); in Compute()
559 OP_REQUIRES_OK(ctx, iterator_resource->SetIteratorFromDataset(ctx, dataset)); in Compute()
1057 IteratorResource* iterator_resource; in Compute() local
1059 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator_resource)); in Compute()
1060 iterator_resource->Unref(); in Compute()
1102 IteratorResource* iterator_resource; in Compute() local
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | range_dataset_serialization_test.py | 40 def _save_op(self, iterator_resource): argument
42 iterator_resource)
48 def _restore_op(self, iterator_resource): argument
52 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | dataset_checkpoint_test.py | 49 def _save_op(self, iterator_resource): argument
51 iterator_resource)
57 def _restore_op(self, iterator_resource): argument
60 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
|
| D | iterator_test.py | 771 def _save_op(iterator_resource): argument
773 iterator_resource)
778 def _restore_op(iterator_resource): argument
781 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.-iterator.pbtxt | 24 …argspec: "args=[\'self\', \'iterator_resource\', \'initializer\', \'output_types\', \'output_shape…
|