| D | nn_ops.py | 4595 … def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
4702 def infer_shape(self, var_shape, m_shape, v_shape, lr_shape, beta1_shape, beta2_shape, argument
4815 def infer_shape(self, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
4955 … def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
5106 … def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
5329 def infer_shape(self, var_shape, accum_shape, lr_shape, l1_shape, l2_shape, argument
5631 def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, lr_shape, argument
5640 lr_shp_len = len(lr_shape)
5643 validator.check_int(lr_shape[0], 1, Rel.EQ, "lr_shape[0]", self.name)
5764 def infer_shape(self, var_shape, accum_shape, accum_update_shape, lr_shape, rho_shape, argument
[all …]
|