Searched defs:offset2bag_data (Results 1 – 1 of 1) sorted by relevance
1078 auto* offset2bag_data = offset2bag.data_ptr<index_t>(); in embedding_bag_cpu_max_out() local1543 auto* offset2bag_data = offset2bag.const_data_ptr<index_t>(); in _embedding_bag_dense_backward_cpu_sum_mean() local1708 auto* offset2bag_data = offset2bag_.const_data_ptr<index_t>(); in _embedding_bag_per_sample_weights_backward_cpu_template() local