Searched full:_infer_size (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _builtins.py | 96 (torch._C._infer_size, "aten::_infer_size"),
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 40 - aten::_infer_size
727 aten::_infer_size: 7
|
| H A D | model_ops.yaml | 36 aten::_infer_size: 9
|
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | kl.py | 420 combined_batch_shape = torch._C._infer_size(
453 combined_batch_shape = torch._C._infer_size(
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_special_ops.cpp | 322 TORCH_SELECTIVE_SCHEMA("aten::_infer_size(int[] a, int[] b) -> int[]"),
|
| /aosp_15_r20/external/pytorch/ |
| H A D | pt_ops.bzl | 582 "aten::_infer_size",
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | Module.cpp | 1327 {"_infer_size", THPModule_inferSize, METH_VARARGS, nullptr},
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 10 from torch._C import _add_docstr, _infer_size
3551 new_size = _infer_size(target.size(), weight.size())
|
| /aosp_15_r20/external/pytorch/torch/_C/ |
| H A D | __init__.pyi.in | 1113 def _infer_size(arg1: Size, arg2: Size) -> Size: ... # THPModule_inferSize
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 655 "torch._C._infer_size",
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_aotdispatch.py | 6307 torch.nn.BCELoss, # new_size = _infer_size(target.size(), weight.size())
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 13351 from torch._C import _infer_size
13355 return _infer_size(x.size(), y.size())
|
| H A D | test_linalg.py | 4329 expand_dims = tuple(torch._C._infer_size(torch.Size(batch_dims_A),
|