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