Searched defs:shard_num (Results 1 – 2 of 2) sorted by relevance
116 Status GetNextFromShard(OpKernelContext* ctx, int shard_num, in GetNextFromShard()195 void GetNextFromShard(IteratorContext* ctx, int shard_num, in GetNextFromShard()670 int32_t shard_num = tensor_shard_num->scalar<int32>()(); in ComputeAsync() local
42 def __init__(self, shard_num, multi_device_iterator_resource, incarnation_id, argument