Searched refs:shard_num (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 101 Status GetNextFromShard(OpKernelContext* ctx, int shard_num, in GetNextFromShard() argument
125 multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, incarnation_id, in GetNextFromShard()
191 void GetNextFromShard(IteratorContext* ctx, int shard_num, in GetNextFromShard() argument
214 if (!buffer_[shard_num].data.empty()) { in GetNextFromShard()
216 std::swap(elem, buffer_[shard_num].data.front()); in GetNextFromShard()
217 buffer_[shard_num].data.pop_front(); in GetNextFromShard()
219 if (buffer_[shard_num].data.size() == max_buffer_size_ - 1) { in GetNextFromShard()
220 buffer_[shard_num].cond_var.notify_all(); in GetNextFromShard()
227 buffer_[shard_num].callbacks.push_back(std::move(callback)); in GetNextFromShard()
228 buffer_[shard_num].cond_var.notify_all(); in GetNextFromShard()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MultiDeviceIteratorGetNextFromShard.pbtxt | 8 name: "shard_num"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MultiDeviceIteratorGetNextFromShard.pbtxt | 8 name: "shard_num"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MultiDeviceIteratorGetNextFromShard.pbtxt | 10 name: "shard_num"
|
| /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
82 shard_num=shard_num,
|
| /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/go/op/ |
| D | wrappers.go | 8272 func MultiDeviceIteratorGetNextFromShard(scope *Scope, multi_device_iterator tf.Output, shard_num t…
8280 multi_device_iterator, shard_num, incarnation_id,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 8657 Arg<TF_Int32Tensor, [{Integer representing which shard to fetch data for.}]>:$shard_num,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25628 name: "shard_num"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 39822 name: "shard_num"
|