Searched defs:unary_attr (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | mkldnn_lowerings.py | 109 unary_attr, argument 143 unary_attr, argument 429 unary_attr, argument 744 unary_attr, argument
|
H A D | mkldnn_ir.py | 1022 unary_attr, argument
|
H A D | ir.py | 6749 unary_attr, argument 6947 unary_attr, argument 7143 unary_attr, argument 7345 unary_attr, argument
|
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
H A D | mkldnn_fusion.py | 240 pattern, unary_attr, computation_op, lowp_dtype=None argument 474 unary_attr=None, argument 514 unary_attr=None, argument
|
H A D | quantization.py | 309 unary_attr, argument 396 unary_attr, argument
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
H A D | Conv.cpp | 346 std::optional<c10::string_view> unary_attr, in mkldnn_convolution_pointwise_binary() 510 std::optional<c10::string_view> unary_attr, in mkldnn_convolution_pointwise_binary_()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
H A D | qconv.cpp | 1408 std::optional<c10::string_view> unary_attr, in _quantized_convolution_onednn() 1938 std::optional<c10::string_view> unary_attr, in run_pointwise_binary()
|
/aosp_15_r20/external/pytorch/test/quantization/core/ |
H A D | test_quantized_op.py | 54 unary_attr : str = "none" variable in PointwisePostOp
|