| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 52 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
58 self.evaluate(multi_device_iterator.initializer)
67 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
74 self.evaluate(multi_device_iterator.initializer)
76 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
80 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
88 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
93 self.evaluate(multi_device_iterator.initializer)
95 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
99 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
[all …]
|
| D | memory_cleanup_test.py | 78 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
82 self.evaluate(multi_device_iterator.get_next())
83 multi_device_iterator._eager_reset()
98 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
100 self.evaluate(multi_device_iterator.get_next())
101 del multi_device_iterator
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | prefetch_with_slack_test.py | 45 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
47 dataset = multi_device_iterator._dataset # pylint: disable=protected-access
54 self.evaluate(multi_device_iterator.initializer)
56 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
60 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 74 multi_device_iterator = (
81 multi_device_iterator=multi_device_iterator,
253 gen_dataset_ops.multi_device_iterator(
384 def __init__(self, multi_device_iterator, iterators, device, deleter): argument
386 self._multi_device_iterator = multi_device_iterator
399 multi_device_iterator=self._multi_device_iterator,
405 multi_device_iterator=self._multi_device_iterator,
571 multi_device_iterator=self._multi_device_iterator_resource,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MultiDeviceIteratorToStringHandle.pbtxt | 4 name: "multi_device_iterator"
|
| D | MultiDeviceIteratorInit.pbtxt | 8 name: "multi_device_iterator"
|
| D | DeleteMultiDeviceIterator.pbtxt | 4 name: "multi_device_iterator"
|
| D | MultiDeviceIteratorFromStringHandle.pbtxt | 8 name: "multi_device_iterator"
|
| D | MultiDeviceIteratorGetNextFromShard.pbtxt | 4 name: "multi_device_iterator"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MultiDeviceIteratorToStringHandle.pbtxt | 4 name: "multi_device_iterator"
|
| D | MultiDeviceIteratorInit.pbtxt | 8 name: "multi_device_iterator"
|
| D | DeleteMultiDeviceIterator.pbtxt | 4 name: "multi_device_iterator"
|
| D | MultiDeviceIteratorFromStringHandle.pbtxt | 8 name: "multi_device_iterator"
|
| D | MultiDeviceIteratorGetNextFromShard.pbtxt | 4 name: "multi_device_iterator"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MultiDeviceIteratorToStringHandle.pbtxt | 4 name: "multi_device_iterator"
|
| D | api_def_MultiDeviceIteratorFromStringHandle.pbtxt | 10 name: "multi_device_iterator"
|
| D | api_def_DeleteMultiDeviceIterator.pbtxt | 5 name: "multi_device_iterator"
|
| D | api_def_MultiDeviceIteratorInit.pbtxt | 10 name: "multi_device_iterator"
|
| D | api_def_MultiDeviceIteratorGetNextFromShard.pbtxt | 4 name: "multi_device_iterator"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1141 …argspec: "args=[\'multi_device_iterator\', \'iterators\', \'deleter\', \'name\'], varargs=None, ke…
2565 …argspec: "args=[\'multi_device_iterator\', \'shard_num\', \'incarnation_id\', \'output_types\', \'…
2569 …argspec: "args=[\'dataset\', \'multi_device_iterator\', \'max_buffer_size\', \'name\'], varargs=No…
2573 …argspec: "args=[\'multi_device_iterator\', \'name\'], varargs=None, keywords=None, defaults=[\'Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1141 …argspec: "args=[\'multi_device_iterator\', \'iterators\', \'deleter\', \'name\'], varargs=None, ke…
2565 …argspec: "args=[\'multi_device_iterator\', \'shard_num\', \'incarnation_id\', \'output_types\', \'…
2569 …argspec: "args=[\'dataset\', \'multi_device_iterator\', \'max_buffer_size\', \'name\'], varargs=No…
2573 …argspec: "args=[\'multi_device_iterator\', \'name\'], varargs=None, keywords=None, defaults=[\'Non…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 8089 … tf.Output, optional ...MultiDeviceIteratorFromStringHandleAttr) (multi_device_iterator tf.Output)…
8174 func MultiDeviceIteratorToStringHandle(scope *Scope, multi_device_iterator tf.Output) (string_handl…
8181 multi_device_iterator,
8272 func MultiDeviceIteratorGetNextFromShard(scope *Scope, multi_device_iterator tf.Output, shard_num t…
8280 multi_device_iterator, shard_num, incarnation_id,
8888 func DeleteMultiDeviceIterator(scope *Scope, multi_device_iterator tf.Output, iterators []tf.Output…
8895 multi_device_iterator, tf.OutputList(iterators), deleter,
13093 func MultiDeviceIteratorInit(scope *Scope, dataset tf.Output, multi_device_iterator tf.Output, max_…
13100 dataset, multi_device_iterator, max_buffer_size,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 2895 …handle to the multi device iterator to delete.}], [TF_DatasetIteratorFree]>:$multi_device_iterator,
8648 …ourceTensor, [{A MultiDeviceIterator resource.}], [TF_DatasetIteratorAlloc]>:$multi_device_iterator
8656 …iceIterator resource.}], [TF_DatasetIteratorRead, TF_DatasetIteratorWrite]>:$multi_device_iterator,
8674 …ourceTensor, [{A MultiDeviceIteratorResource.}], [TF_DatasetIteratorWrite]>:$multi_device_iterator,
8688 …sourceTensor, [{A MultiDeviceIterator resource.}], [TF_DatasetIteratorRead]>:$multi_device_iterator
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 12248 name: "multi_device_iterator"
25598 name: "multi_device_iterator"
25624 name: "multi_device_iterator"
25660 name: "multi_device_iterator"
25676 name: "multi_device_iterator"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 39792 name: "multi_device_iterator"
39818 name: "multi_device_iterator"
39854 name: "multi_device_iterator"
39870 name: "multi_device_iterator"
|