Searched defs:input_x_bdim (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_vmap/ |
| D | vmap_convolution_ops.py | 116 def vmap_rule(dout_bdim, input_x_bdim, weight_size_bdim): argument
135 def vmap_rule(input_x_bdim, dout_bdim, weight_size_bdim): argument
|
| D | vmap_nn_ops.py | 830 def vmap_rule(input_x_bdim, gamma_bdim, beta_bdim, mean_bdim, variance_bdim, u_monad): argument
1636 def vmap_rule(input_x_bdim): argument
1912 …def vmap_rule(input_x_bdim, grid_bdim, interpolation_mode_bdim, padding_mode_bdim, align_corners_b… argument
|
| D | vmap_grad_nn_ops.py | 677 …def vmap_rule(grad_bdim, input_x_bdim, grid_bdim, interpolation_mode_bdim, padding_mode_bdim, alig… argument
|
| D | vmap_array_ops.py | 745 def vmap_rule(input_x_bdim, indices_bdim, updates_bdim): argument
|