Searched refs:softmax_bias_num_rows (Results 1 – 2 of 2) sorted by relevance
170 matrix.rows = softmax_bias_num_rows(0); in GetSoftmaxBias()281 virtual int softmax_bias_num_rows(int i) const = 0;
153 int softmax_bias_num_rows(int i) const override { in softmax_bias_num_rows() function