コード例 #1
0
def _SeluGradGrad(op, grad):
    selu_x = op.inputs[1]
    return (gen_nn_ops.selu_grad(grad, selu_x),
            array_ops.where(selu_x < 0., grad * op.inputs[0],
                            array_ops.zeros_like(selu_x)))
コード例 #2
0
def _SeluGrad(op, grad):
    return gen_nn_ops.selu_grad(grad, op.outputs[0])
コード例 #3
0
ファイル: nn_grad.py プロジェクト: adit-chandra/tensorflow 
def _SeluGrad(op, grad):
  return gen_nn_ops.selu_grad(grad, op.outputs[0])