Searched defs:offset2bag__ (Results 1 – 1 of 1) sorted by relevance
1507 const Tensor& offset2bag__, in _embedding_bag_dense_backward_cpu_sum_mean()1595 const Tensor &offset2bag__, in _embedding_bag_dense_backward_cpu()