Searched refs:softmax_num_rows (Results 1 – 2 of 2) sorted by relevance
155 matrix.rows = softmax_num_rows(0); in GetSoftmaxMatrix()261 virtual int softmax_num_rows(int i) const = 0;
131 int softmax_num_rows(int i) const override { in softmax_num_rows() function