Searched refs:softmax_sparse_input_preprocessing (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | ParamUtils.h | 9 TORCH_API std::tuple<Tensor, Tensor, int64_t> softmax_sparse_input_preprocessing(
|
| H A D | ParamUtils.cpp | 16 std::tuple<Tensor, Tensor, int64_t> softmax_sparse_input_preprocessing( in softmax_sparse_input_preprocessing() function
|
| H A D | SoftMax.cpp | 541 std::tie(input, output, dim) = softmax_sparse_input_preprocessing( in softmax_sparse_cpu()
558 std::tie(input, output, dim) = softmax_sparse_input_preprocessing( in log_softmax_sparse_cpu()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SoftMax.cu | 593 std::tie(input, output, dim) = softmax_sparse_input_preprocessing( in softmax_sparse_cuda()
610 std::tie(input, output, dim) = softmax_sparse_input_preprocessing( in log_softmax_sparse_cuda()
|