Searched defs:_sparse_softmax (Results 1 – 1 of 1) sorted by relevance
609 Tensor _sparse_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in _sparse_softmax() function623 Tensor _sparse_softmax(const Tensor& self, Dimname dim, std::optional<ScalarType> dtype) { in _sparse_softmax() function