Searched refs:excluding_last_dim (Results 1 – 1 of 1) sorted by relevance
3767 const int excluding_last_dim = in Softmax() local3774 for (int i = 0; i < excluding_last_dim; ++i) { in Softmax()3909 const int excluding_last_dim = in SoftmaxInt8LUT() local3947 for (int i = 0; i < excluding_last_dim; ++i) { in SoftmaxInt8LUT()4119 const int excluding_last_dim = in LogSoftmax() local4131 for (int i = 0; i < excluding_last_dim; ++i) { in LogSoftmax()