Searched defs:conv2d_weight (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/jit/ | ||
| D | test_optimize_for_mobile_preserve_debug_info.py | 75 conv2d_weight, argument |
| /external/pytorch/torch/nn/ | ||
| D | grad.py | 156 def conv2d_weight( function |