Lines Matching full:softmax

143   // Returns true if a softmax layer exists.
148   // Returns weight matrix for the softmax layer.  Note: should be called only
153     SAFTM_CHECK(HasSoftmax()) << "No softmax layer.";  in GetSoftmaxMatrix()
164   // Returns bias for the softmax layer.  Technically a Matrix, but we expect it
168     SAFTM_CHECK(HasSoftmax()) << "No softmax layer.";  in GetSoftmaxBias()
255   // ** Access methods for optional MatrixParams softmax.
257   // Returns 1 if proto has optional field softmax, 0 otherwise.
260   // Returns number of rows of transpose(proto.softmax()).
263   // Returns number of columns of transpose(proto.softmax()).
266   // Returns quantization mode for the softmax weights.
271   // Returns pointer to elements of transpose(proto.softmax()), in row-major