Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/
Dmath_grad.cc868 Status MatMulGradCommon(const string& opname, const string& attr_adj_x, in MatMulGradCommon() function
899 return MatMulGradCommon("MatMul", "transpose_a", "transpose_b", attrs, g, in MatMulGrad()
905 return MatMulGradCommon("BatchMatMul", "adj_x", "adj_y", attrs, g, in BatchMatMulGrad()
911 return MatMulGradCommon("BatchMatMulV2", "adj_x", "adj_y", attrs, g, in BatchMatMulV2Grad()
Dmath_grad_test.cc284 void MatMulGradCommon(const string& opname, const string& attr_adj_x, in MatMulGradCommon() function in tensorflow::__anona51524880111::MathGradTest
346 return MatMulGradCommon("MatMul", "transpose_a", "transpose_b", x, ax, y, in MatMulGrad()
352 return MatMulGradCommon("BatchMatMul", "adj_x", "adj_y", x, ax, y, ay, dx, in BatchMatMulGrad()
358 return MatMulGradCommon("BatchMatMulV2", "adj_x", "adj_y", x, ax, y, ay, dx, in BatchMatMulV2Grad()
/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad.cc1106 Status MatMulGradCommon(const Scope& scope, const Operation& op, in MatMulGradCommon() function
1142 return MatMulGradCommon(scope, op, false, grad_inputs, "transpose_a", in MatMulGrad()
1150 return MatMulGradCommon(scope, op, true, grad_inputs, "adj_x", "adj_y", in BatchMatMulGrad()