Searched refs:incarnation_id (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 70 int64* incarnation_id) { in Init() argument
84 *incarnation_id = incarnation_id_; in Init()
93 int64 incarnation_id, in GetNextFromShard() argument
104 multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, incarnation_id, in GetNextFromShard()
135 MultiDeviceBuffer(size_t size, int64 max_buffer_size, int64 incarnation_id, in MultiDeviceBuffer() argument
141 incarnation_id_(incarnation_id), in MultiDeviceBuffer()
173 int64 incarnation_id, in GetNextFromShard() argument
176 if (incarnation_id_ != incarnation_id) { in GetNextFromShard()
514 int64 incarnation_id; in Compute() local
516 &incarnation_id)); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 37 def __init__(self, shard_num, multi_device_iterator_resource, incarnation_id, argument
78 incarnation_id=incarnation_id,
101 if arg == incarnation_id:
154 def __init__(self, per_device_dataset, incarnation_id): argument
166 per_device_dataset._incarnation_id_index] = 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 | 7432 …(scope *Scope, multi_device_iterator tf.Output, shard_num tf.Output, incarnation_id tf.Output, out…
7440 multi_device_iterator, shard_num, incarnation_id,
7466 …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 | 2065 …argspec: "args=[\'multi_device_iterator\', \'shard_num\', \'incarnation_id\', \'output_types\', \'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2065 …argspec: "args=[\'multi_device_iterator\', \'shard_num\', \'incarnation_id\', \'output_types\', \'…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 20108 name: "incarnation_id"
20144 name: "incarnation_id"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 40489 name: "incarnation_id"
40525 name: "incarnation_id"
|
| D | ops_history.v1.pbtxt | 40986 name: "incarnation_id"
41022 name: "incarnation_id"
|