Searched defs:KLDivLoss (Results 1 – 2 of 2) sorted by relevance
156 void KLDivLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y, T *… in KLDivLoss() function
5347 class KLDivLoss(PrimitiveWithInfer): class