Searched refs:gx_1 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 379 std::vector<Output>* grad_outputs, const Output& gx_1, in BinaryGradCommon() argument
384 auto dx_1 = Reshape(scope, Sum(scope, gx_1, rx.r0), sx_1); in BinaryGradCommon()
396 auto gx_1 = Identity(scope, grad_inputs[0]); in AddGrad() local
398 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in AddGrad()
409 auto gx_1 = Identity(scope, grad_inputs[0]); in SubGrad() local
411 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in SubGrad()
423 auto gx_1 = Mul(scope, grad_inputs[0], x_2); in MulGrad() local
425 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in MulGrad()
437 auto gx_1 = Div(scope, grad_inputs[0], x_2); in DivGrad() local
440 return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); in DivGrad()
[all …]
|