Searched defs:offset2bag_arg (Results 1 – 1 of 1) sorted by relevance
1428 auto offset2bag_arg = TensorArg(offset2bag, "offset2bag", 1); in _embedding_bag_backward_symint() local1684 auto offset2bag_arg = TensorArg(offset2bag, "offset2bag", 1); in _embedding_bag_per_sample_weights_backward_cpu_template() local