Searched defs:binary_fn (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/masked/maskedtensor/ |
| D | binary.py | 163 def binary_fn(*args, **kwargs): function
172 def binary_fn(*args, **kwargs): function
|
| /external/pytorch/aten/src/ATen/native/ |
| D | BinaryOps.h | 51 using binary_fn = void(*)(TensorIterator&); variable
|
| D | TensorFactories.h | 137 using binary_fn = void (*)(TensorIterator&); variable
|
| /external/pytorch/test/ |
| D | test_mkldnn_fusion.py | 266 … def __init__(self, binary_fn, dim, in_channels, out_channels, dilation, groups, bias, **kwargs): argument
314 def __init__(self, binary_fn, in_channels, out_channels, bias, **kwargs): argument
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | mkldnn_fusion.py | 193 def _binary_fusion_v1(computation_call, binary_fn): argument
197 def _binary_fusion_v2(computation_call, binary_fn): argument
|
| /external/pytorch/test/inductor/ |
| D | test_mkldnn_pattern_matcher.py | 503 binary_fn, argument
582 def __init__(self, binary_fn, in_channels, out_channels, bias, **kwargs): argument
|