| /external/pytorch/torch/_inductor/ |
| D | mkldnn_lowerings.py | 107 binary_attr, argument
123 binary_attr,
141 binary_attr, argument
157 binary_attr,
427 binary_attr, argument
434 binary_attr == "sum"
463 binary_attr,
742 binary_attr, argument
752 if len(x_size) > 2 and binary_attr == "add":
784 if binary_attr == "sum":
[all …]
|
| D | mkldnn_ir.py | 357 binary_attr: str,
374 binary_attr,
453 binary_attr: str,
470 binary_attr,
912 binary_attr,
950 binary_attr,
1020 binary_attr, argument
1070 binary_attr,
1079 binary_attr == "sum"
1577 binary_attr,
[all …]
|
| D | ir.py | 6114 binary_attr: str,
6131 binary_attr,
6209 binary_attr: str,
6226 binary_attr,
6877 binary_attr,
6901 binary_attr,
6945 binary_attr, argument
6991 binary_attr,
6999 binary_attr == "sum"
7272 binary_attr,
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_kernel_to_llvm_pass.cc | 169 auto binary_attr = in matchAndRewrite() local
171 if (!binary_attr) { in matchAndRewrite()
182 binary_attr.getValue(), LLVM::Linkage::Internal); in matchAndRewrite()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Conv.cpp | 344 c10::string_view binary_attr, in mkldnn_convolution_pointwise_binary() argument
395 auto it_binary = fusion_binary_alg_map().find(binary_attr); in mkldnn_convolution_pointwise_binary()
471 if (binary_attr == "add" && unary_attr_value != "none") { in mkldnn_convolution_pointwise_binary()
475 if (binary_attr == "add") { in mkldnn_convolution_pointwise_binary()
477 } else if (binary_attr == "sub") { in mkldnn_convolution_pointwise_binary()
479 } else if (binary_attr == "mul") { in mkldnn_convolution_pointwise_binary()
508 c10::string_view binary_attr, in mkldnn_convolution_pointwise_binary_() argument
518 binary_attr == "add", in mkldnn_convolution_pointwise_binary_()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qconv.cpp | 1406 std::optional<c10::string_view> binary_attr, in _quantized_convolution_onednn() argument
1433 bool has_binary_post_op = binary_attr.has_value() && binary_attr.value() != "none"; in _quantized_convolution_onednn()
1436 bool has_accum_postop_sum = has_binary_post_op && binary_attr.value() == "sum"; in _quantized_convolution_onednn()
1458 func_name += binary_attr.value().data(); in _quantized_convolution_onednn()
1608 binary_attr.has_value() ? binary_attr.value() : "none", in _quantized_convolution_onednn()
1936 c10::string_view binary_attr, in run_pointwise_binary() argument
1944 act.dim() == 4 && binary_attr == "sum" && ( in run_pointwise_binary()
1953 binary_attr, in run_pointwise_binary()
1963 /*output_dtype*/output_dtype, binary_attr, alpha, in run_pointwise_binary()
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | mkldnn_fusion.py | 482 binary_attr = _binary_attr[binary_op]
484 computation_args = [args_list[0], other] + args_list[1:-3] + [binary_attr]
526 binary_attr = _binary_attr[binary_op]
528 computation_args = [args_list[0], other] + args_list[1:-3] + [binary_attr]
|
| /external/pytorch/test/quantization/core/ |
| D | test_quantized_op.py | 52 binary_attr : str = "none" variable in PointwisePostOp
6708 if post_op.binary_attr == "sum":
6777 if post_op.binary_attr == "sum":
6801 post_op.binary_attr,
7265 pointwise_post_op = PointwisePostOp(binary_attr="sum")
7319 pointwise_post_op = PointwisePostOp(binary_attr="sum", unary_attr="relu")
7375 PointwisePostOp(binary_attr="sum", unary_attr="relu")
7377 else PointwisePostOp(binary_attr="sum")
|