Searched defs:components_require_grad (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nestedtensor.py | 4149 self, device, dtype, keepdim, requires_grad, components_require_grad, func argument
4239 components_require_grad, argument
4308 self, device, dtype, keepdim, requires_grad, components_require_grad, func argument
4350 self, device, dtype, requires_grad, components_require_grad argument
4390 self, device, dtype, requires_grad, components_require_grad argument
4421 self, device, dtype, requires_grad, components_require_grad argument
4478 components_require_grad, argument
4516 components_require_grad, argument
4562 components_require_grad, argument
4622 components_require_grad, argument
[all …]
|