Searched defs:x_bdim (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ | ||
| H A D | custom_op_db.py | 188 def expand_bdim(x, x_bdim): argument |
| H A D | autograd_function_db.py | 323 def expand_bdim(x, x_bdim): argument |