Searched refs:SoftmaxGrad (Results 1 – 3 of 3) sorted by relevance
25 Status SoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in SoftmaxGrad() function48 REGISTER_OP_GRADIENT("Softmax", SoftmaxGrad);
27 Status SoftmaxGrad(const Scope& scope, const Operation& op, in SoftmaxGrad() function48 REGISTER_GRADIENT_OP("Softmax", SoftmaxGrad);
108 TEST_F(NNGradTest, SoftmaxGrad) { in TEST_F() argument