Searched defs:shard_num (Results 1 – 2 of 2) sorted by relevance
92 void GetNextFromShard(OpKernelContext* ctx, int shard_num, in GetNextFromShard()172 void GetNextFromShard(IteratorContext* ctx, int shard_num, in GetNextFromShard()537 int32 shard_num = tensor_shard_num->scalar<int32>()(); in ComputeAsync() local
37 def __init__(self, shard_num, multi_device_iterator_resource, incarnation_id, argument