Searched refs:gx_2 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 380 const Output& gx_2) { in BinaryGradCommon() argument
385 auto dx_2 = Reshape(scope, Sum(scope, gx_2, rx.r1), sx_2); in BinaryGradCommon()
397 auto gx_2 = Identity(scope, grad_inputs[0]); in AddGrad() local
398 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in AddGrad()
410 auto gx_2 = Neg(scope, grad_inputs[0]); in SubGrad() local
411 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in SubGrad()
424 auto gx_2 = Mul(scope, grad_inputs[0], x_1); in MulGrad() local
425 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in MulGrad()
438 auto gx_2 = Mul(scope, grad_inputs[0], in DivGrad() local
440 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in DivGrad()
[all …]
|