| D | _grad_ops.py | 733 def infer_dtype(self, y_backprop_dtype, x_dtype, y_dtype): argument
737 (y_backprop_dtype, x_dtype, y_dtype)))
997 def infer_dtype(self, x_dtype, y_dtype, grad_dtype): argument
998 args = {'x_dtype': x_dtype, 'y_dtype': y_dtype, 'grad_dtype': grad_dtype}
1028 def infer_dtype(self, x_dtype, y_dtype, grad_dtype): argument
1029 args = {'x_dtype': x_dtype, 'y_dtype': y_dtype}
1270 def infer_dtype(self, y_dtype, dy_dtype, dhy_dtype, dcy_dtype, w_dtype, argument
1308 def infer_dtype(self, x_dtype, hx_dtype, y_dtype, reserve_dtype, state_dtype): argument
1361 …def infer_dtype(self, x_dtype, hx_dtype, cx_dtype, w_dtype, y_dtype, hy_dtype, cy_dtype, dy_dtype,… argument
1407 def infer_dtype(self, x_dtype, w_dtype, b_dtype, y_dtype, init_h_dtype, init_c_dtype, h_dtype, argument
[all …]
|