Searched refs:shard_num (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 92 void GetNextFromShard(OpKernelContext* ctx, int shard_num, in GetNextFromShard() argument
104 multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, incarnation_id, in GetNextFromShard()
172 void GetNextFromShard(IteratorContext* ctx, int shard_num, in GetNextFromShard() argument
193 if (!buffer_[shard_num].data.empty()) { in GetNextFromShard()
195 std::swap(elem, buffer_[shard_num].data.front()); in GetNextFromShard()
196 buffer_[shard_num].data.pop_front(); in GetNextFromShard()
198 if (buffer_[shard_num].data.size() == max_buffer_size_ - 1) { in GetNextFromShard()
199 buffer_[shard_num].cond_var.notify_all(); in GetNextFromShard()
206 buffer_[shard_num].callbacks.push_back(std::move(callback)); in GetNextFromShard()
537 int32 shard_num = tensor_shard_num->scalar<int32>()(); in ComputeAsync() local
[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
77 shard_num=shard_num,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MultiDeviceIteratorGetNextFromShard.pbtxt | 10 name: "shard_num"
|
| /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/go/op/ |
| D | wrappers.go | 7432 func MultiDeviceIteratorGetNextFromShard(scope *Scope, multi_device_iterator tf.Output, shard_num t…
7440 multi_device_iterator, shard_num, incarnation_id,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 20104 name: "shard_num"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 40485 name: "shard_num"
|
| D | ops_history.v1.pbtxt | 40982 name: "shard_num"
|