Check the type of inputs.
(beta1, beta2, eps, prim_name)
| 51 | |
| 52 | |
| 53 | def _check_param_value(beta1, beta2, eps, prim_name): |
| 54 | """Check the type of inputs.""" |
| 55 | validator.check_value_type("beta1", beta1, [float], prim_name) |
| 56 | validator.check_value_type("beta2", beta2, [float], prim_name) |
| 57 | validator.check_value_type("eps", eps, [float], prim_name) |
| 58 | validator.check_float_range(beta1, 0.0, 1.0, Rel.INC_NEITHER, "beta1", prim_name) |
| 59 | validator.check_float_range(beta2, 0.0, 1.0, Rel.INC_NEITHER, "beta2", prim_name) |
| 60 | validator.check_positive_float(eps, "eps", prim_name) |
| 61 | |
| 62 | |
| 63 | class AdamWeightDecayOp(Optimizer): |