Home
last modified time | relevance | path

Searched defs:incarnation_id (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc79 int64* incarnation_id) { in Init()
102 int64 incarnation_id, in GetNextFromShard()
154 MultiDeviceBuffer(size_t size, int64 max_buffer_size, int64 incarnation_id, in MultiDeviceBuffer()
192 int64 incarnation_id, in GetNextFromShard()
580 int64 incarnation_id; in Compute() local
609 int64 incarnation_id = tensor_incarnation_id->scalar<int64>()(); in ComputeAsync() local
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py41 def __init__(self, shard_num, multi_device_iterator_resource, incarnation_id, argument
158 def __init__(self, per_device_dataset, incarnation_id): argument
193 def _create_device_dataset(prototype_ds, incarnation_id, prefetch_buffer_size, argument