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