Searched defs:_softmax (Results 1 – 7 of 7) sorted by relevance
33 def _softmax(self, x): member in RaggedSoftmaxTest
93 def _softmax(x): # softmax over 4 dim matrix function
358 def _softmax(func, *args, **kwargs): function
40 TORCH_META_FUNC(_softmax) in TORCH_META_FUNC() argument
98 def _softmax(self, x): member in SoftmaxTest
1201 def _softmax(x: Tensor, dim: int, half_to_float: bool): function
3416 torch::Tensor _softmax(const torch::Tensor& x) { in _softmax() function