Searched defs:softmax_layer (Results 1 – 1 of 1) sorted by relevance
110 SimpleTensor<T> softmax_layer(const SimpleTensor<T> &src, float beta, int32_t axis, bool is_log) in softmax_layer() function116 SimpleTensor<T> softmax_layer(const SimpleTensor<T> &src, float beta, int32_t axis, bool is_log) in softmax_layer() function