Searched refs:n_elements_tensor (Results 1 – 1 of 1) sorted by relevance
2794 def forward(self, x, y, n_elements_tensor): argument2796 n_elements_symint = n_elements_tensor.item()