Searched refs:softmax_nested (Results 1 – 2 of 2) sorted by relevance
52 …y viewing the NestedTensor as a dense tensor. An example of this is `softmax_nested` over ragged d…
504 Tensor softmax_nested( in softmax_nested() function