Home
last modified time | relevance | path

Searched refs:has_torch_function_variadic (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py25 has_torch_function_variadic,
476 if has_torch_function_variadic(input, _random_samples):
521 if has_torch_function_variadic(input, _random_samples):
595 if has_torch_function_variadic(input, _random_samples):
637 if has_torch_function_variadic(input, _random_samples):
2515 if has_torch_function_variadic(input, weight):
2646 if has_torch_function_variadic(input, weight, offsets, per_sample_weights):
2796 if has_torch_function_variadic(input, running_mean, running_var, weight, bias):
2851 if has_torch_function_variadic(input, running_mean, running_var, weight, bias):
2890 if has_torch_function_variadic(input, weight, bias):
[all …]
H A Dinit.py162 if torch.overrides.has_torch_function_variadic(tensor):
189 if torch.overrides.has_torch_function_variadic(tensor):
238 if torch.overrides.has_torch_function_variadic(tensor):
499 if torch.overrides.has_torch_function_variadic(tensor):
/aosp_15_r20/external/pytorch/torch/
H A Dfunctional.py16 has_torch_function_variadic,
1314 if has_torch_function_variadic(a, b):
1473 if has_torch_function_variadic(x1, x2):
H A D_tensor.py26 has_torch_function_variadic,
796 if has_torch_function_variadic(self, other):
961 if has_torch_function_variadic(self, tensor):
H A Doverrides.py245 torch.nn.functional.has_torch_function_variadic,
1788 has_torch_function_variadic = _add_docstr( variable
/aosp_15_r20/external/pytorch/torch/jit/
H A D_script.pyi44 has_torch_function_variadic as has_torch_function_variadic,
H A D_script.py48 has_torch_function_variadic,
1727 _register_builtin(has_torch_function_variadic, "aten::has_torch_function")
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch.py484 torch.overrides.has_torch_function_variadic,
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py1177 ), torch.overrides.has_torch_function_variadic(x)