Searched defs:LogSoftmax (Results 1 – 10 of 10) sorted by relevance
35 LogSoftmax() : PrimitiveC(kNameLogSoftmax) { InitIOName({"x"}, {"output"}); } in LogSoftmax() function
22 class LogSoftmax(Expander): class
43 class LogSoftmax(nn.Cell): class
45 class LogSoftmax(nn.Cell): class
26 class LogSoftmax(nn.Cell): class
46 void LogSoftmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter … in LogSoftmax() function
98 class LogSoftmax(Cell): class
826 MS_TRT_CONVERTER_FUNC_REG(LogSoftmax) { in MS_TRT_CONVERTER_FUNC_REG() argument
441 TEST_F(TestConvert, LogSoftmax) { in TEST_F() argument
290 class LogSoftmax(Primitive): class