| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 75 class MultiDeviceIterator : public ResourceBase { class
77 MultiDeviceIterator( in MultiDeviceIterator() function in tensorflow::data::__anon56eeea570111::MultiDeviceIterator
99 ~MultiDeviceIterator() override { in ~MultiDeviceIterator()
179 MultiDeviceIterator* parent) in MultiDeviceBuffer()
292 &MultiDeviceIterator::MultiDeviceBuffer::BackgroundThread, in EnsureBackgroundThreadStarted()
453 MultiDeviceIterator* const parent_; // Not owned.
498 ->template Delete<MultiDeviceIterator>(cinfo_.container(), in ~MultiDeviceIteratorHandleOp()
522 MultiDeviceIterator* resource; in Compute()
528 resource = new MultiDeviceIterator( in Compute()
534 OP_REQUIRES_OK(context, mgr->Create<MultiDeviceIterator>( in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MultiDeviceIterator.pbtxt | 2 graph_op_name: "MultiDeviceIterator"
4 name: "data.MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 40 multi_device_iterator_ops.MultiDeviceIterator),
139 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
152 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
170 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
188 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
206 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
224 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
253 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
273 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
288 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
[all …]
|
| D | memory_cleanup_test.py | 84 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
101 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MultiDeviceIteratorToStringHandle.pbtxt | 6 A MultiDeviceIterator resource.
15 summary: "Produces a string handle for the given MultiDeviceIterator."
|
| D | api_def_MultiDeviceIteratorFromStringHandle.pbtxt | 12 A MultiDeviceIterator resource.
27 summary: "Generates a MultiDeviceIterator resource from its provided string handle."
|
| D | api_def_MultiDeviceIteratorGetNextFromShard.pbtxt | 6 A MultiDeviceIterator resource.
18 Which incarnation of the MultiDeviceIterator is running.
|
| D | api_def_MultiDeviceIterator.pbtxt | 2 graph_op_name: "MultiDeviceIterator"
41 summary: "Creates a MultiDeviceIterator resource."
|
| D | api_def_AnonymousMultiDeviceIteratorV3.pbtxt | 8 "MultiDeviceIteratorGetNextFromShard" op. In contrast to MultiDeviceIterator,
|
| D | api_def_MultiDeviceIteratorInit.pbtxt | 24 An int64 indicating which incarnation of the MultiDeviceIterator
|
| D | api_def_AnonymousMultiDeviceIterator.pbtxt | 8 "MultiDeviceIteratorGetNextFromShard" op. In contrast to MultiDeviceIterator,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MultiDeviceIterator.pbtxt | 2 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MultiDeviceIterator.pbtxt | 2 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | input_lib.py | 331 self._iterator = multi_device_iterator_ops.MultiDeviceIterator(
338 self._iterator = multi_device_iterator_ops.MultiDeviceIterator(
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | prefetch_with_slack_test.py | 42 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_factory_ops_test.py | 114 it = multi_device_iterator_ops.MultiDeviceIterator(ragged_ds, ['GPU:0'])
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 219 class MultiDeviceIterator: class
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 438 "MultiDeviceIteratorGetNextFromShard" op. In contrast to MultiDeviceIterator,
456 "MultiDeviceIteratorGetNextFromShard" op. In contrast to MultiDeviceIterator,
9602 def TF_MultiDeviceIteratorOp : TF_Op<"MultiDeviceIterator", []> {
9603 let summary = "Creates a MultiDeviceIterator resource.";
9620 Generates a MultiDeviceIterator resource from its provided string handle.
9631 …Res<TF_ResourceTensor, [{A MultiDeviceIterator resource.}], [TF_DatasetIteratorAlloc]>:$multi_devi…
9639 …Arg<TF_ResourceTensor, [{A MultiDeviceIterator resource.}], [TF_DatasetIteratorRead, TF_DatasetIte…
9641 … Arg<TF_Int64Tensor, [{Which incarnation of the MultiDeviceIterator is running.}]>:$incarnation_id
9662 Res<TF_Int64Tensor, [{An int64 indicating which incarnation of the MultiDeviceIterator
9668 let summary = "Produces a string handle for the given MultiDeviceIterator.";
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2620 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2620 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 26813 func MultiDeviceIterator(scope *Scope, devices []string, shared_name string, container string, outp… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 28138 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 39752 name: "MultiDeviceIterator"
|