Searched refs:DiagPartGrad (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc131 Status DiagPartGrad(const Scope& scope, const Operation& op, function in namespace:tensorflow::ops::__anon25351
137 REGISTER_GRADIENT_OP("DiagPart", DiagPartGrad);
H A Darray_grad_test.cc119 TEST_F(ArrayGradTest, DiagPartGrad) {

Completed in 332 milliseconds