Searched refs:LogSoftmaxGrad (Results 1 – 3 of 3) sorted by relevance
50 Status LogSoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in LogSoftmaxGrad() function73 REGISTER_OP_GRADIENT("LogSoftmax", LogSoftmaxGrad);
116 Status LogSoftmaxGrad(const Scope& scope, const Operation& op, in LogSoftmaxGrad() function126 REGISTER_GRADIENT_OP("LogSoftmax", LogSoftmaxGrad);
129 TEST_F(NNGradTest, LogSoftmaxGrad) { in TEST_F() argument