| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 52 class MultiDeviceIterator : public ResourceBase { class
54 MultiDeviceIterator( in MultiDeviceIterator() function in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
156 MultiDeviceIterator* parent) in MultiDeviceBuffer()
248 &MultiDeviceIterator::MultiDeviceBuffer::BackgroundThread, in EnsureBackgroundThreadStarted()
372 MultiDeviceIterator* const parent_; // Not owned.
413 ->template Delete<MultiDeviceIterator>(cinfo_.container(), in ~MultiDeviceIteratorHandleOp()
438 MultiDeviceIterator* resource; in Compute()
444 resource = new MultiDeviceIterator( in Compute()
450 OP_REQUIRES_OK(context, mgr->Create<MultiDeviceIterator>( in Compute()
456 mgr->LookupOrCreate<MultiDeviceIterator>( in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 52 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
67 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
88 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
107 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
131 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
154 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
188 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
215 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
240 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
251 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
[all …]
|
| D | memory_cleanup_test.py | 78 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
98 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
|
| /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/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_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/data/experimental/kernel_tests/ |
| D | prefetch_with_slack_test.py | 45 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 211 class MultiDeviceIterator(object): class
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 1826 self._iterator = multi_device_iterator_ops.MultiDeviceIterator(
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 360 "MultiDeviceIteratorGetNextFromShard" op. In contrast to MultiDeviceIterator,
8619 def TF_MultiDeviceIteratorOp : TF_Op<"MultiDeviceIterator", []> {
8620 let summary = "Creates a MultiDeviceIterator resource.";
8637 Generates a MultiDeviceIterator resource from its provided string handle.
8648 …Res<TF_ResourceTensor, [{A MultiDeviceIterator resource.}], [TF_DatasetIteratorAlloc]>:$multi_devi…
8656 …Arg<TF_ResourceTensor, [{A MultiDeviceIterator resource.}], [TF_DatasetIteratorRead, TF_DatasetIte…
8658 … Arg<TF_Int64Tensor, [{Which incarnation of the MultiDeviceIterator is running.}]>:$incarnation_id
8679 Res<TF_Int64Tensor, [{An int64 indicating which incarnation of the MultiDeviceIterator
8685 let summary = "Produces a string handle for the given MultiDeviceIterator.";
8688 …Arg<TF_ResourceTensor, [{A MultiDeviceIterator resource.}], [TF_DatasetIteratorRead]>:$multi_devic…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2556 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2556 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 8309 func MultiDeviceIterator(scope *Scope, devices []string, shared_name string, container string, outp… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25558 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 39752 name: "MultiDeviceIterator"
|