Searched defs:_log_softmax (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/ |
| H A D | loss.py | 127 def _log_softmax(x, dim, half_to_float, mesh, mesh_dim): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | SoftMax.cpp | 59 TORCH_META_FUNC(_log_softmax) ( in TORCH_META_FUNC() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_test.py | 192 def _log_softmax(self, x): member in LogSoftmaxTest
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 1224 def _log_softmax(x: Tensor, dim: int, half_to_float: bool): function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 2240 def _log_softmax(g: jit_utils.GraphContext, input, dim, half_to_float): function
|