Searched refs:DiagGrad (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ | ||
| H A D | array_grad.cc | 123 Status DiagGrad(const Scope& scope, const Operation& op, function in namespace:tensorflow::ops::__anon25351 129 REGISTER_GRADIENT_OP("Diag", DiagGrad); |
| H A D | array_grad_test.cc | 111 TEST_F(ArrayGradTest, DiagGrad) { |
Completed in 215 milliseconds