Searched refs:softmax_num_cols (Results 1 – 2 of 2) sorted by relevance
156 matrix.cols = softmax_num_cols(0); in GetSoftmaxMatrix()264 virtual int softmax_num_cols(int i) const = 0;
136 int softmax_num_cols(int i) const override { in softmax_num_cols() function