Searched defs:LogSoftmax (Results 1 – 8 of 8) sorted by relevance
22 class LogSoftmax : public OpDesc { class24 LogSoftmax() { in LogSoftmax() function in mindspore::graphkernel::expanders::LogSoftmax
26 class LogSoftmax(nn.Cell): class
50 class LogSoftmax(nn.Cell): class
46 void LogSoftmax(const float *input_ptr, float *output_ptr, float *sum_data, int32_t *input_shape, i… in LogSoftmax() function
282 class LogSoftmax(Cell): class
1437 LogSoftmax() : BaseOperator(kNameLogSoftmax) {} in LogSoftmax() function
837 MS_TRT_CONVERTER_FUNC_REG(LogSoftmax) { in MS_TRT_CONVERTER_FUNC_REG() argument