Home
last modified time | relevance | path

Searched refs:SoftmaxGrad (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dnn_grad.cc25 Status SoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in SoftmaxGrad() function
48 REGISTER_OP_GRADIENT("Softmax", SoftmaxGrad);
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc27 Status SoftmaxGrad(const Scope& scope, const Operation& op, in SoftmaxGrad() function
48 REGISTER_GRADIENT_OP("Softmax", SoftmaxGrad);
Dnn_grad_test.cc108 TEST_F(NNGradTest, SoftmaxGrad) { in TEST_F() argument