Searched defs:unary_fn (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mkldnn_fusion.py | 104 def __init__(self, unary_fn, in_channels, out_channels, bias, **kwargs): argument
196 def __init__(self, unary_fn, in_channels, out_channels, bias, **kwargs): argument
230 … def __init__(self, unary_fn, dim, in_channels, out_channels, dilation, groups, bias, **kwargs): argument
347 def __init__(self, unary_fn, dim, in_channels, out_channels, kernel_size, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/masked/maskedtensor/ |
| H A D | unary.py | 154 def unary_fn(*args, **kwargs): function
163 def unary_fn(*args, **kwargs): function
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_mkldnn_pattern_matcher.py | 118 def _check_unary_is_decomposed(self, unary_fn): argument
271 unary_fn, argument
344 unary_fn, argument
408 def __init__(self, dtype, unary_fn): argument
450 unary_fn, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UnaryOps.h | 16 using unary_fn = void(*)(TensorIteratorBase&); variable
|