| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 43 class MultiDeviceIterator : public ResourceBase { class
45 MultiDeviceIterator( in MultiDeviceIterator() function in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator
137 MultiDeviceIterator* parent) in MultiDeviceBuffer()
226 &MultiDeviceIterator::MultiDeviceBuffer::BackgroundThread, in EnsureBackgroundThreadStarted()
349 MultiDeviceIterator* const parent_; // Not owned.
390 ->template Delete<MultiDeviceIterator>(cinfo_.container(), in ~MultiDeviceIteratorHandleOp()
415 MultiDeviceIterator* resource; in Compute()
421 resource = new MultiDeviceIterator( in Compute()
427 OP_REQUIRES_OK(context, mgr->Create<MultiDeviceIterator>( in Compute()
433 mgr->LookupOrCreate<MultiDeviceIterator>( in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 45 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
56 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
75 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
94 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
118 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
141 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
175 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
202 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
225 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
236 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
[all …]
|
| D | memory_cleanup_test.py | 75 multi_device_iterator = multi_device_iterator_ops.MultiDeviceIterator(
95 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
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 190 class MultiDeviceIterator(object): class
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 390 self._iterator = multi_device_iterator_ops.MultiDeviceIterator(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2056 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2056 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 13147 func MultiDeviceIterator(scope *Scope, devices []string, shared_name string, container string, outp… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 20034 name: "MultiDeviceIterator"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 40415 name: "MultiDeviceIterator"
|
| D | ops_history.v1.pbtxt | 40912 name: "MultiDeviceIterator"
|