Home
last modified time | relevance | path

Searched defs:_softmax (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_math_ops_test.py33 def _softmax(self, x): member in RaggedSoftmaxTest
/external/pytorch/test/nn/
Dtest_multihead_attention.py93 def _softmax(x): # softmax over 4 dim matrix function
/external/pytorch/torch/masked/maskedtensor/
D_ops_refs.py358 def _softmax(func, *args, **kwargs): function
/external/pytorch/aten/src/ATen/native/
DSoftMax.cpp40 TORCH_META_FUNC(_softmax) in TORCH_META_FUNC() argument
/external/tensorflow/tensorflow/python/ops/
Dnn_test.py98 def _softmax(self, x): member in SoftmaxTest
/external/pytorch/torch/_decomp/
Ddecompositions.py1201 def _softmax(x: Tensor, dim: int, half_to_float: bool): function
/external/pytorch/test/cpp/api/
Dmodules.cpp3416 torch::Tensor _softmax(const torch::Tensor& x) { in _softmax() function