Searched defs:offset2bag_ (Results 1 – 1 of 1) sorted by relevance
1413 Tensor offset2bag_; in _embedding_bag_backward_symint() local1517 Tensor &offset2bag_ = const_cast<Tensor &>(offset2bag__); in _embedding_bag_dense_backward_cpu_sum_mean() local1675 Tensor offset2bag_; in _embedding_bag_per_sample_weights_backward_cpu_template() local