| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 79 int64* incarnation_id) { in Init() argument
93 *incarnation_id = incarnation_id_; in Init()
102 int64 incarnation_id, in GetNextFromShard() argument
125 multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, incarnation_id, in GetNextFromShard()
154 MultiDeviceBuffer(size_t size, int64 max_buffer_size, int64 incarnation_id, in MultiDeviceBuffer() argument
160 incarnation_id_(incarnation_id), in MultiDeviceBuffer()
192 int64 incarnation_id, in GetNextFromShard() argument
195 if (incarnation_id_ != incarnation_id) { in GetNextFromShard()
197 "Invalid incarnation id. Provided: ", incarnation_id, in GetNextFromShard()
580 int64 incarnation_id; in Compute() local
[all …]
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 41 def __init__(self, shard_num, multi_device_iterator_resource, incarnation_id, argument
83 incarnation_id=incarnation_id,
106 if arg is incarnation_id:
158 def __init__(self, per_device_dataset, incarnation_id): argument
169 per_device_dataset._incarnation_id_index] = incarnation_id
193 def _create_device_dataset(prototype_ds, incarnation_id, prefetch_buffer_size, argument
196 ds = _ReincarnatedPerDeviceGenerator(prototype_ds, incarnation_id)
541 incarnation_id = gen_dataset_ops.multi_device_iterator_init(
550 incarnation_id, source_device_tensor,
564 incarnation_id, prefetch_buffer_size,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MultiDeviceIteratorInit.pbtxt | 16 name: "incarnation_id"
|
| D | MultiDeviceIteratorGetNextFromShard.pbtxt | 12 name: "incarnation_id"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MultiDeviceIteratorInit.pbtxt | 16 name: "incarnation_id"
|
| D | MultiDeviceIteratorGetNextFromShard.pbtxt | 12 name: "incarnation_id"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MultiDeviceIteratorInit.pbtxt | 22 name: "incarnation_id"
|
| D | api_def_MultiDeviceIteratorGetNextFromShard.pbtxt | 16 name: "incarnation_id"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 8272 …(scope *Scope, multi_device_iterator tf.Output, shard_num tf.Output, incarnation_id tf.Output, out…
8280 multi_device_iterator, shard_num, incarnation_id,
13093 …tf.Output, multi_device_iterator tf.Output, max_buffer_size tf.Output) (incarnation_id tf.Output) {
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2565 …argspec: "args=[\'multi_device_iterator\', \'shard_num\', \'incarnation_id\', \'output_types\', \'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2565 …argspec: "args=[\'multi_device_iterator\', \'shard_num\', \'incarnation_id\', \'output_types\', \'…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 8658 … Arg<TF_Int64Tensor, [{Which incarnation of the MultiDeviceIterator is running.}]>:$incarnation_id
8680 is running.}]>:$incarnation_id
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25632 name: "incarnation_id"
25668 name: "incarnation_id"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 39826 name: "incarnation_id"
39862 name: "incarnation_id"
|