Searched defs:num_bags (Results 1 – 2 of 2) sorted by relevance
31 def __init__(self, num_bags, num_embeddings_per_bag, num_dims): argument
926 auto num_bags = offsets.size(0) - (include_last_offset ? 1 : 0); in make_bag_size_out() local