Home
last modified time | relevance | path

Searched refs:MatrixDiagGrad (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc157 Status MatrixDiagGrad(const Scope& scope, const Operation& op, in MatrixDiagGrad() function
163 REGISTER_GRADIENT_OP("MatrixDiag", MatrixDiagGrad);
Darray_grad_test.cc135 TEST_F(ArrayGradTest, MatrixDiagGrad) { in TEST_F() argument