diff --git a/iree_tests/.gitignore b/iree_tests/.gitignore index 19fbb8e85..1395c179b 100644 --- a/iree_tests/.gitignore +++ b/iree_tests/.gitignore @@ -1,3 +1,7 @@ +# TODO(scotttodd): compile into a build/temp dir instead of the source dir *.vmfb -# TODO(scotttodd): compile into a build dir instead of the source dir +# TODO(scotttodd): convert into a build/temp dir instead of the source dir +*.onnx + +# Leftovers from an older iteration of compile/run test scripts. config_*.txt diff --git a/iree_tests/configs/config_cpu_llvm_sync.json b/iree_tests/configs/config_cpu_llvm_sync.json index 5745e1b8c..b02ccc53a 100644 --- a/iree_tests/configs/config_cpu_llvm_sync.json +++ b/iree_tests/configs/config_cpu_llvm_sync.json @@ -11,8 +11,7 @@ "test_dequantizelinear", "test_slice_default_axes", ], - "skip_run_tests": [ - ], + "skip_run_tests": [], "expected_compile_failures": [ "test_acos_example", "test_acos", @@ -134,7 +133,6 @@ "test_castlike_FLOAT8E5M2FNUZ_to_FLOAT", "test_castlike_STRING_to_FLOAT_expanded", "test_castlike_STRING_to_FLOAT", - "test_celu_expanded", "test_celu", "test_center_crop_pad_crop_and_pad_expanded", "test_center_crop_pad_crop_and_pad", @@ -148,9 +146,7 @@ "test_center_crop_pad_crop", "test_center_crop_pad_pad_expanded", "test_center_crop_pad_pad", - "test_clip_default_inbounds_expanded", "test_clip_default_inbounds", - "test_clip_default_int8_inbounds_expanded", "test_clip_default_int8_inbounds", "test_clip_default_int8_max", "test_clip_default_max", @@ -231,8 +227,6 @@ "test_gemm_default_no_bias", "test_globalmaxpool_precomputed", "test_globalmaxpool", - "test_greater_bcast", - "test_greater", "test_gridsample_aligncorners_true", "test_gridsample_bicubic_align_corners_0_additional_1", "test_gridsample_bicubic_align_corners_1_additional_1", @@ -371,7 +365,6 @@ "test_mod_uint8", "test_momentum_multiple", "test_momentum", - "test_mvn_expanded", "test_mvn", "test_nesterov_momentum", "test_nllloss_NC", @@ -711,13 +704,6 @@ "test_slice_negative_axes", "test_slice_start_out_of_bounds", "test_slice", - "test_softmax_axis_0_expanded", - "test_softmax_axis_1_expanded", - "test_softmax_axis_2_expanded", - "test_softmax_default_axis_expanded", - "test_softmax_example_expanded", - "test_softmax_large_number_expanded", - "test_softmax_negative_axis_expanded", "test_softplus_example", "test_softplus", "test_softsign_example", diff --git a/iree_tests/configs/config_gpu_vulkan.json b/iree_tests/configs/config_gpu_vulkan.json index a8ad1582c..888dcaf10 100644 --- a/iree_tests/configs/config_gpu_vulkan.json +++ b/iree_tests/configs/config_gpu_vulkan.json @@ -10,7 +10,6 @@ // Timeouts? "test_dequantizelinear", "test_slice_default_axes", - "test_sum_example", ], "skip_run_tests": [], "expected_compile_failures": [ @@ -134,7 +133,6 @@ "test_castlike_FLOAT8E5M2FNUZ_to_FLOAT", "test_castlike_STRING_to_FLOAT_expanded", "test_castlike_STRING_to_FLOAT", - "test_celu_expanded", "test_celu", "test_center_crop_pad_crop_and_pad_expanded", "test_center_crop_pad_crop_and_pad", @@ -148,9 +146,7 @@ "test_center_crop_pad_crop", "test_center_crop_pad_pad_expanded", "test_center_crop_pad_pad", - "test_clip_default_inbounds_expanded", "test_clip_default_inbounds", - "test_clip_default_int8_inbounds_expanded", "test_clip_default_int8_inbounds", "test_clip_default_int8_max", "test_clip_default_max", @@ -231,8 +227,6 @@ "test_gemm_default_no_bias", "test_globalmaxpool_precomputed", "test_globalmaxpool", - "test_greater_bcast", - "test_greater", "test_gridsample_aligncorners_true", "test_gridsample_bicubic_align_corners_0_additional_1", "test_gridsample_bicubic_align_corners_1_additional_1", @@ -371,7 +365,6 @@ "test_mod_uint8", "test_momentum_multiple", "test_momentum", - "test_mvn_expanded", "test_mvn", "test_nesterov_momentum", "test_nllloss_NC", @@ -711,13 +704,6 @@ "test_slice_negative_axes", "test_slice_start_out_of_bounds", "test_slice", - "test_softmax_axis_0_expanded", - "test_softmax_axis_1_expanded", - "test_softmax_axis_2_expanded", - "test_softmax_default_axis_expanded", - "test_softmax_example_expanded", - "test_softmax_large_number_expanded", - "test_softmax_negative_axis_expanded", "test_softplus_example", "test_softplus", "test_softsign_example", @@ -941,6 +927,7 @@ "test_castlike_FLOAT_to_FLOAT16", "test_castlike_FLOAT16_to_FLOAT_expanded", "test_castlike_FLOAT16_to_FLOAT", + "test_clip_default_int8_inbounds_expanded", "test_isinf_float16", "test_isnan_float16", "test_max_float16", diff --git a/iree_tests/onnx/import_tests.py b/iree_tests/onnx/import_tests.py index 9f6b37e24..135fe7a52 100644 --- a/iree_tests/onnx/import_tests.py +++ b/iree_tests/onnx/import_tests.py @@ -4,9 +4,11 @@ # See https://llvm.org/LICENSE.txt for license information. # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +import argparse import onnx +from multiprocessing import Pool from pathlib import Path -from onnx import numpy_helper +from onnx import numpy_helper, version_converter import shutil import subprocess import numpy as np @@ -20,6 +22,9 @@ ONNX_REPO_GENERATED_TESTS_ROOT = ONNX_REPO_ROOT / "onnx/backend/test/data" NODE_TESTS_ROOT = ONNX_REPO_GENERATED_TESTS_ROOT / "node" +# Convert test cases to at least this version using The ONNX Version Converter. +ONNX_CONVERTER_OUTPUT_MIN_VERSION = 17 + # Write imported files to our own 'generated' folder. GENERATED_FILES_OUTPUT_ROOT = REPO_ROOT / "iree_tests/onnx/node/generated" @@ -48,6 +53,16 @@ def convert_io_proto(proto_filename, type_proto): return None +def import_onnx_files_with_cleanup(test_dir_path): + test_name = test_dir_path.name + imported_dir_path = Path(GENERATED_FILES_OUTPUT_ROOT) / test_name + result = import_onnx_files(test_dir_path, imported_dir_path) + if not result: + # Note: could comment this out to keep partially imported directories. + shutil.rmtree(imported_dir_path) + return (test_name, result) + + def import_onnx_files(test_dir_path, imported_dir_path): # This imports one 'test_[name]' subfolder from this: # @@ -71,15 +86,32 @@ def import_onnx_files(test_dir_path, imported_dir_path): test_data_flagfile_path = imported_dir_path / "test_data_flags.txt" test_data_flagfile_lines = [] - # Import model.onnx to model.mlir. - # TODO(scotttodd): copy the .onnx file into the generated folder? Useful for reproducing - # could also add a symlink or other files with info - # e.g. importer tool / version / flags used, reproducer command - onnx_model_path = test_dir_path / "model.onnx" + # Convert model.onnx up to ONNX_CONVERTER_OUTPUT_MIN_VERSION if needed. + # TODO(scotttodd): stamp some info e.g. importer tool / version / flags used + original_model_path = test_dir_path / "model.onnx" + converted_model_path = imported_dir_path / "model.onnx" + + original_model = onnx.load_model(original_model_path) + original_version = original_model.opset_import[0].version + if original_version < ONNX_CONVERTER_OUTPUT_MIN_VERSION: + try: + converted_model = version_converter.convert_version( + original_model, ONNX_CONVERTER_OUTPUT_MIN_VERSION + ) + onnx.save(converted_model, converted_model_path) + except: + # Conversion failed. Do our best with the original file. + # print(f"WARNING: ONNX conversion failed for {test_dir_path.name}") + shutil.copy(original_model_path, converted_model_path) + else: + # No conversion needed. + shutil.copy(original_model_path, converted_model_path) + + # Import converted model.onnx to model.mlir. imported_model_path = imported_dir_path / "model.mlir" exec_args = [ "iree-import-onnx", - str(onnx_model_path), + str(converted_model_path), "-o", str(imported_model_path), ] @@ -101,7 +133,7 @@ def import_onnx_files(test_dir_path, imported_dir_path): test_data_dir = test_data_dirs[0] test_inputs = list(test_data_dir.glob("input_*.pb")) test_outputs = list(test_data_dir.glob("output_*.pb")) - model = onnx.load(onnx_model_path) + model = onnx.load(converted_model_path) for i in range(len(test_inputs)): test_input = test_inputs[i] t = convert_io_proto(test_input, model.graph.input[i].type) @@ -126,6 +158,16 @@ def import_onnx_files(test_dir_path, imported_dir_path): if __name__ == "__main__": + parser = argparse.ArgumentParser(description="ONNX test case importer.") + parser.add_argument( + "-j", + "--jobs", + type=int, + default=8, + help="Number of parallel processes to use when importing test cases", + ) + args = parser.parse_args() + test_dir_paths = find_onnx_tests(NODE_TESTS_ROOT) # TODO(scotttodd): add flag to not clear output dir? @@ -134,27 +176,16 @@ def import_onnx_files(test_dir_path, imported_dir_path): GENERATED_FILES_OUTPUT_ROOT.mkdir(parents=True) print(f"Importing tests in '{NODE_TESTS_ROOT}'") - print("******************************************************************") passed_imports = [] failed_imports = [] - # TODO(scotttodd): parallelize this (or move into a test runner like pytest) - for i in range(len(test_dir_paths)): - test_dir_path = test_dir_paths[i] - test_name = test_dir_path.name - - current_number = str(i).rjust(4, "0") - progress_str = f"[{current_number}/{len(test_dir_paths)}]" - print(f"{progress_str}: Importing {test_name}") - - imported_dir_path = Path(GENERATED_FILES_OUTPUT_ROOT) / test_name - result = import_onnx_files(test_dir_path, imported_dir_path) - if result: - passed_imports.append(test_name) - else: - failed_imports.append(test_name) - # Note: could comment this out to keep partially imported directories. - shutil.rmtree(imported_dir_path) + with Pool(args.jobs) as pool: + results = pool.imap_unordered(import_onnx_files_with_cleanup, test_dir_paths) + for result in results: + if result[1]: + passed_imports.append(result[0]) + else: + failed_imports.append(result[0]) print("******************************************************************") passed_imports.sort() diff --git a/iree_tests/onnx/node/generated/test_abs/model.mlir b/iree_tests/onnx/node/generated/test_abs/model.mlir index fa9c85ba3..b45a36ce0 100644 --- a/iree_tests/onnx/node/generated/test_abs/model.mlir +++ b/iree_tests/onnx/node/generated/test_abs/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_abs(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_abs(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Abs"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_acos/model.mlir b/iree_tests/onnx/node/generated/test_acos/model.mlir index d2f753394..b00af0f1b 100644 --- a/iree_tests/onnx/node/generated/test_acos/model.mlir +++ b/iree_tests/onnx/node/generated/test_acos/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_acos(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_acos(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Acos"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_acos_example/model.mlir b/iree_tests/onnx/node/generated/test_acos_example/model.mlir index 7df5b8d0b..e2d19c890 100644 --- a/iree_tests/onnx/node/generated/test_acos_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_acos_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_acos_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_acos_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Acos"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_acosh/model.mlir b/iree_tests/onnx/node/generated/test_acosh/model.mlir index 178669ba6..13ac8813a 100644 --- a/iree_tests/onnx/node/generated/test_acosh/model.mlir +++ b/iree_tests/onnx/node/generated/test_acosh/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_acosh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_acosh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Acosh"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_acosh_example/model.mlir b/iree_tests/onnx/node/generated/test_acosh_example/model.mlir index 36c324258..6f0529119 100644 --- a/iree_tests/onnx/node/generated/test_acosh_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_acosh_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_acosh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_acosh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Acosh"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_add/model.mlir b/iree_tests/onnx/node/generated/test_add/model.mlir index 582adb766..d0e4a50c2 100644 --- a/iree_tests/onnx/node/generated/test_add/model.mlir +++ b/iree_tests/onnx/node/generated/test_add/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_add(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_add(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Add"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_add_bcast/model.mlir b/iree_tests/onnx/node/generated/test_add_bcast/model.mlir index 611c1d700..bbf7ce52b 100644 --- a/iree_tests/onnx/node/generated/test_add_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_add_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_add_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_add_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Add"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_add_uint8/model.mlir b/iree_tests/onnx/node/generated/test_add_uint8/model.mlir index 3b5a57fa7..19f2544b9 100644 --- a/iree_tests/onnx/node/generated/test_add_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_add_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_add_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_add_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Add"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],ui8>, !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> return %0 : !torch.vtensor<[3,4,5],ui8> diff --git a/iree_tests/onnx/node/generated/test_and2d/model.mlir b/iree_tests/onnx/node/generated/test_and2d/model.mlir index 9327923db..1079ce0de 100644 --- a/iree_tests/onnx/node/generated/test_and2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and2d(%arg0: !torch.vtensor<[3,4],i1>, %arg1: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and2d(%arg0: !torch.vtensor<[3,4],i1>, %arg1: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[3,4],i1>, !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> return %0 : !torch.vtensor<[3,4],i1> diff --git a/iree_tests/onnx/node/generated/test_and3d/model.mlir b/iree_tests/onnx/node/generated/test_and3d/model.mlir index 3706d003d..65f5f656e 100644 --- a/iree_tests/onnx/node/generated/test_and3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and3d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and3d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_and4d/model.mlir b/iree_tests/onnx/node/generated/test_and4d/model.mlir index b5b31b9e3..b37dddd81 100644 --- a/iree_tests/onnx/node/generated/test_and4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and4d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and4d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_and_bcast3v1d/model.mlir b/iree_tests/onnx/node/generated/test_and_bcast3v1d/model.mlir index cedb0256a..e0cc72098 100644 --- a/iree_tests/onnx/node/generated/test_and_bcast3v1d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and_bcast3v1d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and_bcast3v1d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and_bcast3v1d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_and_bcast3v2d/model.mlir b/iree_tests/onnx/node/generated/test_and_bcast3v2d/model.mlir index 4d67a5700..8c0ed44f6 100644 --- a/iree_tests/onnx/node/generated/test_and_bcast3v2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and_bcast3v2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and_bcast3v2d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and_bcast3v2d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_and_bcast4v2d/model.mlir b/iree_tests/onnx/node/generated/test_and_bcast4v2d/model.mlir index be4c6736c..4fef9fa89 100644 --- a/iree_tests/onnx/node/generated/test_and_bcast4v2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and_bcast4v2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and_bcast4v2d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and_bcast4v2d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_and_bcast4v3d/model.mlir b/iree_tests/onnx/node/generated/test_and_bcast4v3d/model.mlir index b4d6bc00b..cea5a4865 100644 --- a/iree_tests/onnx/node/generated/test_and_bcast4v3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and_bcast4v3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and_bcast4v3d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and_bcast4v3d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_and_bcast4v4d/model.mlir b/iree_tests/onnx/node/generated/test_and_bcast4v4d/model.mlir index b292e2b51..bce784493 100644 --- a/iree_tests/onnx/node/generated/test_and_bcast4v4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_and_bcast4v4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_and_bcast4v4d(%arg0: !torch.vtensor<[1,4,1,6],i1>, %arg1: !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_and_bcast4v4d(%arg0: !torch.vtensor<[1,4,1,6],i1>, %arg1: !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.And"(%arg0, %arg1) : (!torch.vtensor<[1,4,1,6],i1>, !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_argmax_default_axis_example/model.mlir b/iree_tests/onnx/node/generated/test_argmax_default_axis_example/model.mlir index 61a56f925..432d52cce 100644 --- a/iree_tests/onnx/node/generated/test_argmax_default_axis_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_default_axis_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_default_axis_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_default_axis_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> return %0 : !torch.vtensor<[1,2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_default_axis_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_default_axis_example_select_last_index/model.mlir index e372f9768..6ade7e02a 100644 --- a/iree_tests/onnx/node/generated/test_argmax_default_axis_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_default_axis_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_default_axis_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_default_axis_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> return %0 : !torch.vtensor<[1,2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_default_axis_random/model.mlir b/iree_tests/onnx/node/generated/test_argmax_default_axis_random/model.mlir index 39fa5f4ab..7042cfe84 100644 --- a/iree_tests/onnx/node/generated/test_argmax_default_axis_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_default_axis_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_default_axis_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_default_axis_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> return %0 : !torch.vtensor<[1,3,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_default_axis_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_default_axis_random_select_last_index/model.mlir index 4181c8146..08fffc803 100644 --- a/iree_tests/onnx/node/generated/test_argmax_default_axis_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_default_axis_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_default_axis_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_default_axis_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> return %0 : !torch.vtensor<[1,3,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_argmax_keepdims_example/model.mlir index 4c9d15c3a..8944bce63 100644 --- a/iree_tests/onnx/node/generated/test_argmax_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_keepdims_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_keepdims_example_select_last_index/model.mlir index 22fad46b9..fe79bb2c1 100644 --- a/iree_tests/onnx/node/generated/test_argmax_keepdims_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_keepdims_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_argmax_keepdims_random/model.mlir index 695870bfa..986206b7e 100644 --- a/iree_tests/onnx/node/generated/test_argmax_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> return %0 : !torch.vtensor<[2,1,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_keepdims_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_keepdims_random_select_last_index/model.mlir index 3127da817..5b116941d 100644 --- a/iree_tests/onnx/node/generated/test_argmax_keepdims_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_keepdims_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> return %0 : !torch.vtensor<[2,1,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example/model.mlir index ea5bf3679..18a87ab50 100644 --- a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_negative_axis_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_negative_axis_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example_select_last_index/model.mlir index 5e2298ec9..feedc14c3 100644 --- a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_negative_axis_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_negative_axis_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random/model.mlir index 6d25c8105..f258400f8 100644 --- a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_negative_axis_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_negative_axis_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> return %0 : !torch.vtensor<[2,3,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random_select_last_index/model.mlir index 90ba8a91b..41c2f254d 100644 --- a/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_negative_axis_keepdims_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_negative_axis_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_negative_axis_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> return %0 : !torch.vtensor<[2,3,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example/model.mlir index eb6178273..b02bfa5bb 100644 --- a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_no_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_no_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> return %0 : !torch.vtensor<[2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example_select_last_index/model.mlir index c4b7cebdd..2220d201b 100644 --- a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_no_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_no_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> return %0 : !torch.vtensor<[2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random/model.mlir index 696f7a0d8..31bcebeb2 100644 --- a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_no_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_no_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> return %0 : !torch.vtensor<[2,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random_select_last_index/model.mlir index 47690039f..de7d2705b 100644 --- a/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmax_no_keepdims_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmax_no_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmax_no_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMax"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> return %0 : !torch.vtensor<[2,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_default_axis_example/model.mlir b/iree_tests/onnx/node/generated/test_argmin_default_axis_example/model.mlir index a7fd6a385..69b0aabd1 100644 --- a/iree_tests/onnx/node/generated/test_argmin_default_axis_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_default_axis_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_default_axis_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_default_axis_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> return %0 : !torch.vtensor<[1,2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_default_axis_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_default_axis_example_select_last_index/model.mlir index 7dac179b4..11d23be58 100644 --- a/iree_tests/onnx/node/generated/test_argmin_default_axis_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_default_axis_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_default_axis_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_default_axis_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[1,2],si64> return %0 : !torch.vtensor<[1,2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_default_axis_random/model.mlir b/iree_tests/onnx/node/generated/test_argmin_default_axis_random/model.mlir index 406d9b310..1b2d8a450 100644 --- a/iree_tests/onnx/node/generated/test_argmin_default_axis_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_default_axis_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_default_axis_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_default_axis_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> return %0 : !torch.vtensor<[1,3,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_default_axis_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_default_axis_random_select_last_index/model.mlir index d9486a1ca..14e6dc947 100644 --- a/iree_tests/onnx/node/generated/test_argmin_default_axis_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_default_axis_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_default_axis_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_default_axis_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[1,3,4],si64> return %0 : !torch.vtensor<[1,3,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_argmin_keepdims_example/model.mlir index 0c47095c1..d25b75849 100644 --- a/iree_tests/onnx/node/generated/test_argmin_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_keepdims_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_keepdims_example_select_last_index/model.mlir index 36513ab58..762a1fc74 100644 --- a/iree_tests/onnx/node/generated/test_argmin_keepdims_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_keepdims_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_argmin_keepdims_random/model.mlir index 993b1806c..9b4fa8751 100644 --- a/iree_tests/onnx/node/generated/test_argmin_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> return %0 : !torch.vtensor<[2,1,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_keepdims_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_keepdims_random_select_last_index/model.mlir index 4f3304e10..b6863e411 100644 --- a/iree_tests/onnx/node/generated/test_argmin_keepdims_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_keepdims_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,1,4],si64> return %0 : !torch.vtensor<[2,1,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example/model.mlir index b578ec9af..9edec48f7 100644 --- a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_negative_axis_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_negative_axis_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example_select_last_index/model.mlir index 3a90c53e5..4eb6778aa 100644 --- a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_negative_axis_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_negative_axis_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,1],si64> return %0 : !torch.vtensor<[2,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random/model.mlir index 908eb2730..ddc205d8f 100644 --- a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_negative_axis_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_negative_axis_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> return %0 : !torch.vtensor<[2,3,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random_select_last_index/model.mlir index 71b0f7a56..70196729f 100644 --- a/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_negative_axis_keepdims_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_negative_axis_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_negative_axis_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.keepdims = 1 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,3,1],si64> return %0 : !torch.vtensor<[2,3,1],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example/model.mlir index c567beb48..a9ec6bb1a 100644 --- a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_no_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_no_keepdims_example(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> return %0 : !torch.vtensor<[2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example_select_last_index/model.mlir index 8b0b31104..768779330 100644 --- a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_example_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_no_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_no_keepdims_example_select_last_index(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2],si64> return %0 : !torch.vtensor<[2],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random/model.mlir index 88ee5c191..12bd9c7d9 100644 --- a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_no_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_no_keepdims_random(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> return %0 : !torch.vtensor<[2,4],si64> diff --git a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random_select_last_index/model.mlir b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random_select_last_index/model.mlir index ae2e43cc4..783d22d0c 100644 --- a/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random_select_last_index/model.mlir +++ b/iree_tests/onnx/node/generated/test_argmin_no_keepdims_random_select_last_index/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_argmin_no_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_argmin_no_keepdims_random_select_last_index(%arg0: !torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ArgMin"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.keepdims = 0 : si64, torch.onnx.select_last_index = 1 : si64} : (!torch.vtensor<[2,3,4],f32>) -> !torch.vtensor<[2,4],si64> return %0 : !torch.vtensor<[2,4],si64> diff --git a/iree_tests/onnx/node/generated/test_asin/model.mlir b/iree_tests/onnx/node/generated/test_asin/model.mlir index b7fb6a5dd..6f4c5acc3 100644 --- a/iree_tests/onnx/node/generated/test_asin/model.mlir +++ b/iree_tests/onnx/node/generated/test_asin/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_asin(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_asin(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Asin"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_asin_example/model.mlir b/iree_tests/onnx/node/generated/test_asin_example/model.mlir index 47896f291..bb07dbee9 100644 --- a/iree_tests/onnx/node/generated/test_asin_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_asin_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_asin_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_asin_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Asin"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_asinh/model.mlir b/iree_tests/onnx/node/generated/test_asinh/model.mlir index 153672353..39ba2efe6 100644 --- a/iree_tests/onnx/node/generated/test_asinh/model.mlir +++ b/iree_tests/onnx/node/generated/test_asinh/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_asinh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_asinh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Asinh"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_asinh_example/model.mlir b/iree_tests/onnx/node/generated/test_asinh_example/model.mlir index f28c55714..eed9df61c 100644 --- a/iree_tests/onnx/node/generated/test_asinh_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_asinh_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_asinh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_asinh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Asinh"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_atan/model.mlir b/iree_tests/onnx/node/generated/test_atan/model.mlir index ea8937a1d..957f6ee0c 100644 --- a/iree_tests/onnx/node/generated/test_atan/model.mlir +++ b/iree_tests/onnx/node/generated/test_atan/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_atan(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_atan(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Atan"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_atan_example/model.mlir b/iree_tests/onnx/node/generated/test_atan_example/model.mlir index 6efe5425e..6b08a5e09 100644 --- a/iree_tests/onnx/node/generated/test_atan_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_atan_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_atan_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_atan_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Atan"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_atanh/model.mlir b/iree_tests/onnx/node/generated/test_atanh/model.mlir index 7ae12f84a..582a95642 100644 --- a/iree_tests/onnx/node/generated/test_atanh/model.mlir +++ b/iree_tests/onnx/node/generated/test_atanh/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_atanh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_atanh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Atanh"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_atanh_example/model.mlir b/iree_tests/onnx/node/generated/test_atanh_example/model.mlir index 8da789a93..292078bc0 100644 --- a/iree_tests/onnx/node/generated/test_atanh_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_atanh_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_atanh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_atanh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Atanh"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_basic_conv_with_padding/model.mlir b/iree_tests/onnx/node/generated/test_basic_conv_with_padding/model.mlir index 29e904ae0..ce3112061 100644 --- a/iree_tests/onnx/node/generated/test_basic_conv_with_padding/model.mlir +++ b/iree_tests/onnx/node/generated/test_basic_conv_with_padding/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_basic_conv_with_padding(%arg0: !torch.vtensor<[1,1,5,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_basic_conv_with_padding(%arg0: !torch.vtensor<[1,1,5,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Conv"(%arg0, %arg1) {torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.pads = [1 : si64, 1 : si64, 1 : si64, 1 : si64]} : (!torch.vtensor<[1,1,5,5],f32>, !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,5,5],f32> return %0 : !torch.vtensor<[1,1,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_basic_conv_without_padding/model.mlir b/iree_tests/onnx/node/generated/test_basic_conv_without_padding/model.mlir index bbb7f1f46..0f6f2d1be 100644 --- a/iree_tests/onnx/node/generated/test_basic_conv_without_padding/model.mlir +++ b/iree_tests/onnx/node/generated/test_basic_conv_without_padding/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_basic_conv_without_padding(%arg0: !torch.vtensor<[1,1,5,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_basic_conv_without_padding(%arg0: !torch.vtensor<[1,1,5,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Conv"(%arg0, %arg1) {torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.pads = [0 : si64, 0 : si64, 0 : si64, 0 : si64]} : (!torch.vtensor<[1,1,5,5],f32>, !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,3],f32> return %0 : !torch.vtensor<[1,1,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_batchnorm_epsilon/model.mlir b/iree_tests/onnx/node/generated/test_batchnorm_epsilon/model.mlir index 11e125564..f0fea4b50 100644 --- a/iree_tests/onnx/node/generated/test_batchnorm_epsilon/model.mlir +++ b/iree_tests/onnx/node/generated/test_batchnorm_epsilon/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_batchnorm_epsilon(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_batchnorm_epsilon(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BatchNormalization"(%arg0, %arg1, %arg2, %arg3, %arg4) {torch.onnx.epsilon = 0.00999999977 : f32} : (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> return %0 : !torch.vtensor<[2,3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_batchnorm_epsilon_training_mode/model.mlir b/iree_tests/onnx/node/generated/test_batchnorm_epsilon_training_mode/model.mlir index d9cb010af..47372d9cf 100644 --- a/iree_tests/onnx/node/generated/test_batchnorm_epsilon_training_mode/model.mlir +++ b/iree_tests/onnx/node/generated/test_batchnorm_epsilon_training_mode/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_batchnorm_epsilon_training_mode(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_batchnorm_epsilon_training_mode(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.BatchNormalization"(%arg0, %arg1, %arg2, %arg3, %arg4) {torch.onnx.epsilon = 0.00999999977 : f32, torch.onnx.training_mode = 1 : si64} : (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) return %0#0, %0#1, %0#2 : !torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_batchnorm_example/model.mlir b/iree_tests/onnx/node/generated/test_batchnorm_example/model.mlir index 9029d2e73..677fc5259 100644 --- a/iree_tests/onnx/node/generated/test_batchnorm_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_batchnorm_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_batchnorm_example(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_batchnorm_example(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BatchNormalization"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> return %0 : !torch.vtensor<[2,3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_batchnorm_example_training_mode/model.mlir b/iree_tests/onnx/node/generated/test_batchnorm_example_training_mode/model.mlir index ecfc04464..120adcbfa 100644 --- a/iree_tests/onnx/node/generated/test_batchnorm_example_training_mode/model.mlir +++ b/iree_tests/onnx/node/generated/test_batchnorm_example_training_mode/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_batchnorm_example_training_mode(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_batchnorm_example_training_mode(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>, %arg3: !torch.vtensor<[3],f32>, %arg4: !torch.vtensor<[3],f32>) -> (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.BatchNormalization"(%arg0, %arg1, %arg2, %arg3, %arg4) {torch.onnx.training_mode = 1 : si64} : (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) return %0#0, %0#1, %0#2 : !torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_bernoulli/model.mlir b/iree_tests/onnx/node/generated/test_bernoulli/model.mlir index 232b9b1ef..a09886b16 100644 --- a/iree_tests/onnx/node/generated/test_bernoulli/model.mlir +++ b/iree_tests/onnx/node/generated/test_bernoulli/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bernoulli(%arg0: !torch.vtensor<[10],f64>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bernoulli(%arg0: !torch.vtensor<[10],f64>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Bernoulli"(%arg0) : (!torch.vtensor<[10],f64>) -> !torch.vtensor<[10],f64> return %0 : !torch.vtensor<[10],f64> diff --git a/iree_tests/onnx/node/generated/test_bernoulli_double/model.mlir b/iree_tests/onnx/node/generated/test_bernoulli_double/model.mlir index 77cd97e33..e461274c0 100644 --- a/iree_tests/onnx/node/generated/test_bernoulli_double/model.mlir +++ b/iree_tests/onnx/node/generated/test_bernoulli_double/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bernoulli_double(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bernoulli_double(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Bernoulli"(%arg0) {torch.onnx.dtype = 11 : si64} : (!torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f64> return %0 : !torch.vtensor<[10],f64> diff --git a/iree_tests/onnx/node/generated/test_bernoulli_double_expanded/model.mlir b/iree_tests/onnx/node/generated/test_bernoulli_double_expanded/model.mlir index 6eef43678..cd2c59cf3 100644 --- a/iree_tests/onnx/node/generated/test_bernoulli_double_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_bernoulli_double_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bernoulli_double_expanded(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bernoulli_double_expanded(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.RandomUniformLike"(%arg0) {torch.onnx.dtype = 1 : si64, torch.onnx.high = 1.000000e+00 : f32, torch.onnx.low = 0.000000e+00 : f32} : (!torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f32> %1 = torch.operator "onnx.Greater"(%0, %arg0) : (!torch.vtensor<[10],f32>, !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],i1> diff --git a/iree_tests/onnx/node/generated/test_bernoulli_expanded/model.mlir b/iree_tests/onnx/node/generated/test_bernoulli_expanded/model.mlir index 6190e9731..c6dc28951 100644 --- a/iree_tests/onnx/node/generated/test_bernoulli_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_bernoulli_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bernoulli_expanded(%arg0: !torch.vtensor<[10],f64>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bernoulli_expanded(%arg0: !torch.vtensor<[10],f64>) -> !torch.vtensor<[10],f64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.RandomUniformLike"(%arg0) {torch.onnx.dtype = 11 : si64, torch.onnx.high = 1.000000e+00 : f32, torch.onnx.low = 0.000000e+00 : f32} : (!torch.vtensor<[10],f64>) -> !torch.vtensor<[10],f64> %1 = torch.operator "onnx.Greater"(%0, %arg0) : (!torch.vtensor<[10],f64>, !torch.vtensor<[10],f64>) -> !torch.vtensor<[10],i1> diff --git a/iree_tests/onnx/node/generated/test_bernoulli_seed/model.mlir b/iree_tests/onnx/node/generated/test_bernoulli_seed/model.mlir index 187fdd97e..61969e15f 100644 --- a/iree_tests/onnx/node/generated/test_bernoulli_seed/model.mlir +++ b/iree_tests/onnx/node/generated/test_bernoulli_seed/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bernoulli_seed(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bernoulli_seed(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Bernoulli"(%arg0) {torch.onnx.seed = 0.000000e+00 : f32} : (!torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f32> return %0 : !torch.vtensor<[10],f32> diff --git a/iree_tests/onnx/node/generated/test_bernoulli_seed_expanded/model.mlir b/iree_tests/onnx/node/generated/test_bernoulli_seed_expanded/model.mlir index 3ca8fd712..3d491a409 100644 --- a/iree_tests/onnx/node/generated/test_bernoulli_seed_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_bernoulli_seed_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bernoulli_seed_expanded(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bernoulli_seed_expanded(%arg0: !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.RandomUniformLike"(%arg0) {torch.onnx.dtype = 1 : si64, torch.onnx.high = 1.000000e+00 : f32, torch.onnx.low = 0.000000e+00 : f32, torch.onnx.seed = 0.000000e+00 : f32} : (!torch.vtensor<[10],f32>) -> !torch.vtensor<[10],f32> %1 = torch.operator "onnx.Greater"(%0, %arg0) : (!torch.vtensor<[10],f32>, !torch.vtensor<[10],f32>) -> !torch.vtensor<[10],i1> diff --git a/iree_tests/onnx/node/generated/test_bitshift_left_uint16/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_left_uint16/model.mlir index 35fdef21e..72dfe8fa5 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_left_uint16/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_left_uint16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_left_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_left_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "LEFT"} : (!torch.vtensor<[3],ui16>, !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> return %0 : !torch.vtensor<[3],ui16> diff --git a/iree_tests/onnx/node/generated/test_bitshift_left_uint32/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_left_uint32/model.mlir index 77b097596..52c662dc0 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_left_uint32/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_left_uint32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_left_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_left_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "LEFT"} : (!torch.vtensor<[3],ui32>, !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> return %0 : !torch.vtensor<[3],ui32> diff --git a/iree_tests/onnx/node/generated/test_bitshift_left_uint64/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_left_uint64/model.mlir index 9f5653086..2dfe56c75 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_left_uint64/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_left_uint64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_left_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_left_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "LEFT"} : (!torch.vtensor<[3],ui64>, !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> return %0 : !torch.vtensor<[3],ui64> diff --git a/iree_tests/onnx/node/generated/test_bitshift_left_uint8/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_left_uint8/model.mlir index 0f01b7f72..d6ba901d2 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_left_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_left_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_left_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_left_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "LEFT"} : (!torch.vtensor<[3],ui8>, !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> return %0 : !torch.vtensor<[3],ui8> diff --git a/iree_tests/onnx/node/generated/test_bitshift_right_uint16/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_right_uint16/model.mlir index 0c2744e14..7237598f5 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_right_uint16/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_right_uint16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_right_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_right_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "RIGHT"} : (!torch.vtensor<[3],ui16>, !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> return %0 : !torch.vtensor<[3],ui16> diff --git a/iree_tests/onnx/node/generated/test_bitshift_right_uint32/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_right_uint32/model.mlir index e05db104c..88f7f750e 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_right_uint32/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_right_uint32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_right_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_right_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "RIGHT"} : (!torch.vtensor<[3],ui32>, !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> return %0 : !torch.vtensor<[3],ui32> diff --git a/iree_tests/onnx/node/generated/test_bitshift_right_uint64/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_right_uint64/model.mlir index 89e124e52..4467612bb 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_right_uint64/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_right_uint64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_right_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_right_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "RIGHT"} : (!torch.vtensor<[3],ui64>, !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> return %0 : !torch.vtensor<[3],ui64> diff --git a/iree_tests/onnx/node/generated/test_bitshift_right_uint8/model.mlir b/iree_tests/onnx/node/generated/test_bitshift_right_uint8/model.mlir index f6fe1edab..8df19a7a8 100644 --- a/iree_tests/onnx/node/generated/test_bitshift_right_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_bitshift_right_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_bitshift_right_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_bitshift_right_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.BitShift"(%arg0, %arg1) {torch.onnx.direction = "RIGHT"} : (!torch.vtensor<[3],ui8>, !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> return %0 : !torch.vtensor<[3],ui8> diff --git a/iree_tests/onnx/node/generated/test_ceil/model.mlir b/iree_tests/onnx/node/generated/test_ceil/model.mlir index 63b5d458b..3d7180b67 100644 --- a/iree_tests/onnx/node/generated/test_ceil/model.mlir +++ b/iree_tests/onnx/node/generated/test_ceil/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_ceil(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_ceil(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Ceil"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_ceil_example/model.mlir b/iree_tests/onnx/node/generated/test_ceil_example/model.mlir index c8400f595..765851a83 100644 --- a/iree_tests/onnx/node/generated/test_ceil_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_ceil_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_ceil_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_ceil_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Ceil"(%arg0) : (!torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> return %0 : !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_celu/model.mlir b/iree_tests/onnx/node/generated/test_celu/model.mlir index 8b6e4f3b3..2baf2c779 100644 --- a/iree_tests/onnx/node/generated/test_celu/model.mlir +++ b/iree_tests/onnx/node/generated/test_celu/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_celu(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_celu(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Celu"(%arg0) {torch.onnx.alpha = 2.000000e+00 : f32} : (!torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> return %0 : !torch.vtensor<[3,3,3,1],f32> diff --git a/iree_tests/onnx/node/generated/test_celu_expanded/model.mlir b/iree_tests/onnx/node/generated/test_celu_expanded/model.mlir index 59e8940d4..b3fe20dbc 100644 --- a/iree_tests/onnx/node/generated/test_celu_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_celu_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_celu_expanded(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_celu_expanded(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2.000000e+00> : tensor<1xf32>} : () -> !torch.vtensor<[1],f32> %1 = torch.operator "onnx.Div"(%arg0, %0) : (!torch.vtensor<[3,3,3,1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3,3,1],f32> diff --git a/iree_tests/onnx/node/generated/test_clip/model.mlir b/iree_tests/onnx/node/generated/test_clip/model.mlir index 28b36bc80..f8ce6b3f9 100644 --- a/iree_tests/onnx/node/generated/test_clip/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_default_inbounds/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_inbounds/model.mlir index 47c727d49..84a5de7e6 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_inbounds/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_inbounds/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_inbounds(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_inbounds(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %none, %none) : (!torch.vtensor<[3],f32>, !torch.none, !torch.none) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_default_inbounds_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_inbounds_expanded/model.mlir index 525511b46..61e4c557d 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_inbounds_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_inbounds_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_inbounds_expanded(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_inbounds_expanded(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Identity"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds/model.mlir index 4d89e1aeb..4e34dace6 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_int8_inbounds(%arg0: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_int8_inbounds(%arg0: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %none, %none) : (!torch.vtensor<[3],si8>, !torch.none, !torch.none) -> !torch.vtensor<[3],si8> return %0 : !torch.vtensor<[3],si8> diff --git a/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds_expanded/model.mlir index 8bd8febf0..d1c0783d6 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_int8_inbounds_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_int8_inbounds_expanded(%arg0: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_int8_inbounds_expanded(%arg0: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Identity"(%arg0) : (!torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> return %0 : !torch.vtensor<[3],si8> diff --git a/iree_tests/onnx/node/generated/test_clip_default_int8_max/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_int8_max/model.mlir index ccbc4d6c0..baa87873d 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_int8_max/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_int8_max/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_int8_max(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_int8_max(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %none, %arg1) : (!torch.vtensor<[3,4,5],si8>, !torch.none, !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> return %0 : !torch.vtensor<[3,4,5],si8> diff --git a/iree_tests/onnx/node/generated/test_clip_default_int8_max_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_int8_max_expanded/model.mlir index 53292d707..d5f6b1dfa 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_int8_max_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_int8_max_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_int8_max_expanded(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_int8_max_expanded(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg1, %arg0) : (!torch.vtensor<[],si8>, !torch.vtensor<[3,4,5],si8>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[],si8>, !torch.vtensor<[3,4,5],si8>) -> !torch.vtensor<[3,4,5],si8> diff --git a/iree_tests/onnx/node/generated/test_clip_default_int8_min/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_int8_min/model.mlir index 5c7a4b959..527e80104 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_int8_min/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_int8_min/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_int8_min(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_int8_min(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],si8>, !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> return %0 : !torch.vtensor<[3,4,5],si8> diff --git a/iree_tests/onnx/node/generated/test_clip_default_int8_min_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_int8_min_expanded/model.mlir index 401688328..1691c5aee 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_int8_min_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_int8_min_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_int8_min_expanded(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_int8_min_expanded(%arg0: !torch.vtensor<[3,4,5],si8>, %arg1: !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],si8>, !torch.vtensor<[],si8>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[],si8>, !torch.vtensor<[3,4,5],si8>) -> !torch.vtensor<[3,4,5],si8> diff --git a/iree_tests/onnx/node/generated/test_clip_default_max/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_max/model.mlir index 8ba5240a5..cbb76decf 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_max/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_max/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_max(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_max(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %none, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.none, !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_default_max_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_max_expanded/model.mlir index b5f70e48e..da059a6ea 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_max_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_max_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_max_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_max_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg1, %arg0) : (!torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_default_min/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_min/model.mlir index e5abe1ef7..3451c8ef4 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_min/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_min/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_min(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_min(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_default_min_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_default_min_expanded/model.mlir index eb2b0205d..dcdfbbb1a 100644 --- a/iree_tests/onnx/node/generated/test_clip_default_min_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_default_min_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_default_min_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_default_min_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_example/model.mlir b/iree_tests/onnx/node/generated/test_clip_example/model.mlir index 5f33e9136..c09ea24fd 100644 --- a/iree_tests/onnx/node/generated/test_clip_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_example_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_example_expanded/model.mlir index 0fa131d18..17feeba9c 100644 --- a/iree_tests/onnx/node/generated/test_clip_example_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_example_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_example_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_example_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3],i1>, !torch.vtensor<[],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_expanded/model.mlir index 64806a97d..a8e642243 100644 --- a/iree_tests/onnx/node/generated/test_clip_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_inbounds/model.mlir b/iree_tests/onnx/node/generated/test_clip_inbounds/model.mlir index ea19764f6..f869e7ca1 100644 --- a/iree_tests/onnx/node/generated/test_clip_inbounds/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_inbounds/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_inbounds(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_inbounds(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_inbounds_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_inbounds_expanded/model.mlir index 35b1d6c59..12e34ed82 100644 --- a/iree_tests/onnx/node/generated/test_clip_inbounds_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_inbounds_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_inbounds_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_inbounds_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3],i1>, !torch.vtensor<[],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_outbounds/model.mlir b/iree_tests/onnx/node/generated/test_clip_outbounds/model.mlir index d4f99ccd6..808074461 100644 --- a/iree_tests/onnx/node/generated/test_clip_outbounds/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_outbounds/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_outbounds(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_outbounds(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_outbounds_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_outbounds_expanded/model.mlir index 037efdf9a..6d6853f6f 100644 --- a/iree_tests/onnx/node/generated/test_clip_outbounds_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_outbounds_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_outbounds_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_outbounds_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3],i1>, !torch.vtensor<[],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_splitbounds/model.mlir b/iree_tests/onnx/node/generated/test_clip_splitbounds/model.mlir index 26834f091..be7657d09 100644 --- a/iree_tests/onnx/node/generated/test_clip_splitbounds/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_splitbounds/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_splitbounds(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_splitbounds(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Clip"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_clip_splitbounds_expanded/model.mlir b/iree_tests/onnx/node/generated/test_clip_splitbounds_expanded/model.mlir index fe932d45a..fa8ebeb58 100644 --- a/iree_tests/onnx/node/generated/test_clip_splitbounds_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_clip_splitbounds_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_clip_splitbounds_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_clip_splitbounds_expanded(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],i1> %1 = torch.operator "onnx.Where"(%0, %arg1, %arg0) : (!torch.vtensor<[3],i1>, !torch.vtensor<[],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_compress_0/model.mlir b/iree_tests/onnx/node/generated/test_compress_0/model.mlir index e16cb201c..abb63438f 100644 --- a/iree_tests/onnx/node/generated/test_compress_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_compress_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_compress_0(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[3],i1>) -> !torch.vtensor<[2,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_compress_0(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[3],i1>) -> !torch.vtensor<[2,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Compress"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[3,2],f32>, !torch.vtensor<[3],i1>) -> !torch.vtensor<[2,2],f32> return %0 : !torch.vtensor<[2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_compress_1/model.mlir b/iree_tests/onnx/node/generated/test_compress_1/model.mlir index 0a730fb91..06328c761 100644 --- a/iree_tests/onnx/node/generated/test_compress_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_compress_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_compress_1(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[2],i1>) -> !torch.vtensor<[3,1],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_compress_1(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[2],i1>) -> !torch.vtensor<[3,1],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Compress"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[3,2],f32>, !torch.vtensor<[2],i1>) -> !torch.vtensor<[3,1],f32> return %0 : !torch.vtensor<[3,1],f32> diff --git a/iree_tests/onnx/node/generated/test_compress_default_axis/model.mlir b/iree_tests/onnx/node/generated/test_compress_default_axis/model.mlir index 73b48236a..da063cccd 100644 --- a/iree_tests/onnx/node/generated/test_compress_default_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_compress_default_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_compress_default_axis(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_compress_default_axis(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Compress"(%arg0, %arg1) : (!torch.vtensor<[3,2],f32>, !torch.vtensor<[5],i1>) -> !torch.vtensor<[2],f32> return %0 : !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_compress_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_compress_negative_axis/model.mlir index 9e44afee0..a4cc865a1 100644 --- a/iree_tests/onnx/node/generated/test_compress_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_compress_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_compress_negative_axis(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[2],i1>) -> !torch.vtensor<[3,1],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_compress_negative_axis(%arg0: !torch.vtensor<[3,2],f32>, %arg1: !torch.vtensor<[2],i1>) -> !torch.vtensor<[3,1],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Compress"(%arg0, %arg1) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[3,2],f32>, !torch.vtensor<[2],i1>) -> !torch.vtensor<[3,1],f32> return %0 : !torch.vtensor<[3,1],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_1d_axis_0/model.mlir b/iree_tests/onnx/node/generated/test_concat_1d_axis_0/model.mlir index bf6247fb0..85f420974 100644 --- a/iree_tests/onnx/node/generated/test_concat_1d_axis_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_1d_axis_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_1d_axis_0(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[2],f32>) -> !torch.vtensor<[4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_1d_axis_0(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[2],f32>) -> !torch.vtensor<[4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[4],f32> return %0 : !torch.vtensor<[4],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_1d_axis_negative_1/model.mlir b/iree_tests/onnx/node/generated/test_concat_1d_axis_negative_1/model.mlir index 52a05dfc3..c95e228ca 100644 --- a/iree_tests/onnx/node/generated/test_concat_1d_axis_negative_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_1d_axis_negative_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_1d_axis_negative_1(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[2],f32>) -> !torch.vtensor<[4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_1d_axis_negative_1(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[2],f32>) -> !torch.vtensor<[4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[4],f32> return %0 : !torch.vtensor<[4],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_2d_axis_0/model.mlir b/iree_tests/onnx/node/generated/test_concat_2d_axis_0/model.mlir index b7fc8d02f..0a903795d 100644 --- a/iree_tests/onnx/node/generated/test_concat_2d_axis_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_2d_axis_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_2d_axis_0(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_2d_axis_0(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[4,2],f32> return %0 : !torch.vtensor<[4,2],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_2d_axis_1/model.mlir b/iree_tests/onnx/node/generated/test_concat_2d_axis_1/model.mlir index 71c00906a..ea0469c7e 100644 --- a/iree_tests/onnx/node/generated/test_concat_2d_axis_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_2d_axis_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_2d_axis_1(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_2d_axis_1(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,4],f32> return %0 : !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_1/model.mlir b/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_1/model.mlir index ad4ce8897..803218110 100644 --- a/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_2d_axis_negative_1(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_2d_axis_negative_1(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,4],f32> return %0 : !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_2/model.mlir b/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_2/model.mlir index adf48cf2f..193523316 100644 --- a/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_2/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_2d_axis_negative_2/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_2d_axis_negative_2(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_2d_axis_negative_2(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = -2 : si64} : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[4,2],f32> return %0 : !torch.vtensor<[4,2],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_3d_axis_0/model.mlir b/iree_tests/onnx/node/generated/test_concat_3d_axis_0/model.mlir index 4e0e3e657..85cd9fb9b 100644 --- a/iree_tests/onnx/node/generated/test_concat_3d_axis_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_3d_axis_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_3d_axis_0(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[4,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_3d_axis_0(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[4,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[2,2,2],f32>, !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[4,2,2],f32> return %0 : !torch.vtensor<[4,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_3d_axis_1/model.mlir b/iree_tests/onnx/node/generated/test_concat_3d_axis_1/model.mlir index d25e5147d..7dd4303cc 100644 --- a/iree_tests/onnx/node/generated/test_concat_3d_axis_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_3d_axis_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_3d_axis_1(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_3d_axis_1(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[2,2,2],f32>, !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,4,2],f32> return %0 : !torch.vtensor<[2,4,2],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_3d_axis_2/model.mlir b/iree_tests/onnx/node/generated/test_concat_3d_axis_2/model.mlir index 23d91addd..08614d30d 100644 --- a/iree_tests/onnx/node/generated/test_concat_3d_axis_2/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_3d_axis_2/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_3d_axis_2(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_3d_axis_2(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[2,2,2],f32>, !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,2,4],f32> return %0 : !torch.vtensor<[2,2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_1/model.mlir b/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_1/model.mlir index c4886a1b4..c0e1bcc7e 100644 --- a/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_3d_axis_negative_1(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_3d_axis_negative_1(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[2,2,2],f32>, !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,2,4],f32> return %0 : !torch.vtensor<[2,2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_2/model.mlir b/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_2/model.mlir index d28a5e362..a04727f82 100644 --- a/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_2/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_2/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_3d_axis_negative_2(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_3d_axis_negative_2(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,4,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = -2 : si64} : (!torch.vtensor<[2,2,2],f32>, !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[2,4,2],f32> return %0 : !torch.vtensor<[2,4,2],f32> diff --git a/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_3/model.mlir b/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_3/model.mlir index faf2c985d..b23ed8116 100644 --- a/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_3/model.mlir +++ b/iree_tests/onnx/node/generated/test_concat_3d_axis_negative_3/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_concat_3d_axis_negative_3(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[4,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_concat_3d_axis_negative_3(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[4,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Concat"(%arg0, %arg1) {torch.onnx.axis = -3 : si64} : (!torch.vtensor<[2,2,2],f32>, !torch.vtensor<[2,2,2],f32>) -> !torch.vtensor<[4,2,2],f32> return %0 : !torch.vtensor<[4,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_conv_with_autopad_same/model.mlir b/iree_tests/onnx/node/generated/test_conv_with_autopad_same/model.mlir index 8257f91b1..6b59c5a11 100644 --- a/iree_tests/onnx/node/generated/test_conv_with_autopad_same/model.mlir +++ b/iree_tests/onnx/node/generated/test_conv_with_autopad_same/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_conv_with_autopad_same(%arg0: !torch.vtensor<[1,1,5,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_conv_with_autopad_same(%arg0: !torch.vtensor<[1,1,5,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Conv"(%arg0, %arg1) {torch.onnx.auto_pad = "SAME_LOWER", torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,5,5],f32>, !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,3],f32> return %0 : !torch.vtensor<[1,1,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_conv_with_strides_and_asymmetric_padding/model.mlir b/iree_tests/onnx/node/generated/test_conv_with_strides_and_asymmetric_padding/model.mlir index 31a72cef5..450675c18 100644 --- a/iree_tests/onnx/node/generated/test_conv_with_strides_and_asymmetric_padding/model.mlir +++ b/iree_tests/onnx/node/generated/test_conv_with_strides_and_asymmetric_padding/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_conv_with_strides_and_asymmetric_padding(%arg0: !torch.vtensor<[1,1,7,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,4,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_conv_with_strides_and_asymmetric_padding(%arg0: !torch.vtensor<[1,1,7,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,4,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Conv"(%arg0, %arg1) {torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.pads = [1 : si64, 0 : si64, 1 : si64, 0 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,7,5],f32>, !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,4,2],f32> return %0 : !torch.vtensor<[1,1,4,2],f32> diff --git a/iree_tests/onnx/node/generated/test_conv_with_strides_no_padding/model.mlir b/iree_tests/onnx/node/generated/test_conv_with_strides_no_padding/model.mlir index c644e6ce5..6ec8abb25 100644 --- a/iree_tests/onnx/node/generated/test_conv_with_strides_no_padding/model.mlir +++ b/iree_tests/onnx/node/generated/test_conv_with_strides_no_padding/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_conv_with_strides_no_padding(%arg0: !torch.vtensor<[1,1,7,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_conv_with_strides_no_padding(%arg0: !torch.vtensor<[1,1,7,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Conv"(%arg0, %arg1) {torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.pads = [0 : si64, 0 : si64, 0 : si64, 0 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,7,5],f32>, !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,3,2],f32> return %0 : !torch.vtensor<[1,1,3,2],f32> diff --git a/iree_tests/onnx/node/generated/test_conv_with_strides_padding/model.mlir b/iree_tests/onnx/node/generated/test_conv_with_strides_padding/model.mlir index e9c3fe21a..90472b8c2 100644 --- a/iree_tests/onnx/node/generated/test_conv_with_strides_padding/model.mlir +++ b/iree_tests/onnx/node/generated/test_conv_with_strides_padding/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_conv_with_strides_padding(%arg0: !torch.vtensor<[1,1,7,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,4,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_conv_with_strides_padding(%arg0: !torch.vtensor<[1,1,7,5],f32>, %arg1: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,4,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Conv"(%arg0, %arg1) {torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.pads = [1 : si64, 1 : si64, 1 : si64, 1 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,7,5],f32>, !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,4,3],f32> return %0 : !torch.vtensor<[1,1,4,3],f32> diff --git a/iree_tests/onnx/node/generated/test_convinteger_with_padding/model.mlir b/iree_tests/onnx/node/generated/test_convinteger_with_padding/model.mlir index f02968f36..f05c9a22e 100644 --- a/iree_tests/onnx/node/generated/test_convinteger_with_padding/model.mlir +++ b/iree_tests/onnx/node/generated/test_convinteger_with_padding/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convinteger_with_padding(%arg0: !torch.vtensor<[1,1,3,3],ui8>, %arg1: !torch.vtensor<[1,1,2,2],ui8>, %arg2: !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,4,4],si32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convinteger_with_padding(%arg0: !torch.vtensor<[1,1,3,3],ui8>, %arg1: !torch.vtensor<[1,1,2,2],ui8>, %arg2: !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,4,4],si32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvInteger"(%arg0, %arg1, %arg2) {torch.onnx.pads = [1 : si64, 1 : si64, 1 : si64, 1 : si64]} : (!torch.vtensor<[1,1,3,3],ui8>, !torch.vtensor<[1,1,2,2],ui8>, !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,4,4],si32> return %0 : !torch.vtensor<[1,1,4,4],si32> diff --git a/iree_tests/onnx/node/generated/test_convinteger_without_padding/model.mlir b/iree_tests/onnx/node/generated/test_convinteger_without_padding/model.mlir index 7b4615d79..6382a610e 100644 --- a/iree_tests/onnx/node/generated/test_convinteger_without_padding/model.mlir +++ b/iree_tests/onnx/node/generated/test_convinteger_without_padding/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convinteger_without_padding(%arg0: !torch.vtensor<[1,1,3,3],ui8>, %arg1: !torch.vtensor<[1,1,2,2],ui8>, %arg2: !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,2,2],si32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convinteger_without_padding(%arg0: !torch.vtensor<[1,1,3,3],ui8>, %arg1: !torch.vtensor<[1,1,2,2],ui8>, %arg2: !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,2,2],si32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvInteger"(%arg0, %arg1, %arg2) : (!torch.vtensor<[1,1,3,3],ui8>, !torch.vtensor<[1,1,2,2],ui8>, !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,2,2],si32> return %0 : !torch.vtensor<[1,1,2,2],si32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose/model.mlir index 4e78286f3..04718b336 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) : (!torch.vtensor<[1,1,3,3],f32>, !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,5,5],f32> return %0 : !torch.vtensor<[1,2,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_1d/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_1d/model.mlir index 3a8d30590..8bc3ddc5b 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_1d/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_1d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_1d(%arg0: !torch.vtensor<[1,1,3],f32>, %arg1: !torch.vtensor<[1,2,3],f32>) -> !torch.vtensor<[1,2,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_1d(%arg0: !torch.vtensor<[1,1,3],f32>, %arg1: !torch.vtensor<[1,2,3],f32>) -> !torch.vtensor<[1,2,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) : (!torch.vtensor<[1,1,3],f32>, !torch.vtensor<[1,2,3],f32>) -> !torch.vtensor<[1,2,5],f32> return %0 : !torch.vtensor<[1,2,5],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_3d/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_3d/model.mlir index bf81efeef..ea771efc1 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_3d(%arg0: !torch.vtensor<[1,1,3,4,5],f32>, %arg1: !torch.vtensor<[1,2,3,3,3],f32>) -> !torch.vtensor<[1,2,5,6,7],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_3d(%arg0: !torch.vtensor<[1,1,3,4,5],f32>, %arg1: !torch.vtensor<[1,2,3,3,3],f32>) -> !torch.vtensor<[1,2,5,6,7],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) : (!torch.vtensor<[1,1,3,4,5],f32>, !torch.vtensor<[1,2,3,3,3],f32>) -> !torch.vtensor<[1,2,5,6,7],f32> return %0 : !torch.vtensor<[1,2,5,6,7],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_autopad_same/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_autopad_same/model.mlir index 2695ff383..f090969f2 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_autopad_same/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_autopad_same/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_autopad_same(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,6,6],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_autopad_same(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,6,6],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) {torch.onnx.auto_pad = "SAME_UPPER", torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,3,3],f32>, !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,6,6],f32> return %0 : !torch.vtensor<[1,2,6,6],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_dilations/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_dilations/model.mlir index 0e73f427e..af63785d7 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_dilations/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_dilations/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_dilations(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,1,2,2],f32>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_dilations(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,1,2,2],f32>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) {torch.onnx.dilations = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,3,3],f32>, !torch.vtensor<[1,1,2,2],f32>) -> !torch.vtensor<[1,1,5,5],f32> return %0 : !torch.vtensor<[1,1,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_kernel_shape/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_kernel_shape/model.mlir index 7d020e864..67a553450 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_kernel_shape/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_kernel_shape/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_kernel_shape(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_kernel_shape(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) {torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.output_padding = [1 : si64, 1 : si64], torch.onnx.output_shape = [10 : si64, 8 : si64], torch.onnx.strides = [3 : si64, 2 : si64]} : (!torch.vtensor<[1,1,3,3],f32>, !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> return %0 : !torch.vtensor<[1,2,10,8],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_output_shape/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_output_shape/model.mlir index 4812fc41b..167504162 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_output_shape/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_output_shape/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_output_shape(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_output_shape(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) {torch.onnx.output_shape = [10 : si64, 8 : si64], torch.onnx.strides = [3 : si64, 2 : si64]} : (!torch.vtensor<[1,1,3,3],f32>, !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> return %0 : !torch.vtensor<[1,2,10,8],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_pad/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_pad/model.mlir index c9acdf7eb..a718dd923 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_pad/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_pad/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_pad(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_pad(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) {torch.onnx.output_padding = [1 : si64, 1 : si64], torch.onnx.strides = [3 : si64, 2 : si64]} : (!torch.vtensor<[1,1,3,3],f32>, !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,10,8],f32> return %0 : !torch.vtensor<[1,2,10,8],f32> diff --git a/iree_tests/onnx/node/generated/test_convtranspose_pads/model.mlir b/iree_tests/onnx/node/generated/test_convtranspose_pads/model.mlir index 1fd779983..b95576bb8 100644 --- a/iree_tests/onnx/node/generated/test_convtranspose_pads/model.mlir +++ b/iree_tests/onnx/node/generated/test_convtranspose_pads/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_convtranspose_pads(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,7,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_convtranspose_pads(%arg0: !torch.vtensor<[1,1,3,3],f32>, %arg1: !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,7,3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ConvTranspose"(%arg0, %arg1) {torch.onnx.pads = [1 : si64, 2 : si64, 1 : si64, 2 : si64], torch.onnx.strides = [3 : si64, 2 : si64]} : (!torch.vtensor<[1,1,3,3],f32>, !torch.vtensor<[1,2,3,3],f32>) -> !torch.vtensor<[1,2,7,3],f32> return %0 : !torch.vtensor<[1,2,7,3],f32> diff --git a/iree_tests/onnx/node/generated/test_cos/model.mlir b/iree_tests/onnx/node/generated/test_cos/model.mlir index 02631402c..5e56507e0 100644 --- a/iree_tests/onnx/node/generated/test_cos/model.mlir +++ b/iree_tests/onnx/node/generated/test_cos/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cos(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cos(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Cos"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_cos_example/model.mlir b/iree_tests/onnx/node/generated/test_cos_example/model.mlir index 73ae1d9d2..3970c6c05 100644 --- a/iree_tests/onnx/node/generated/test_cos_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_cos_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cos_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cos_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Cos"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_cosh/model.mlir b/iree_tests/onnx/node/generated/test_cosh/model.mlir index dc6078302..010aed5c3 100644 --- a/iree_tests/onnx/node/generated/test_cosh/model.mlir +++ b/iree_tests/onnx/node/generated/test_cosh/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cosh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cosh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Cosh"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_cosh_example/model.mlir b/iree_tests/onnx/node/generated/test_cosh_example/model.mlir index 2c436635e..8dd0396a5 100644 --- a/iree_tests/onnx/node/generated/test_cosh_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_cosh_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cosh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cosh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Cosh"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_cumsum_1d/model.mlir b/iree_tests/onnx/node/generated/test_cumsum_1d/model.mlir index 9067d8f26..c75342daf 100644 --- a/iree_tests/onnx/node/generated/test_cumsum_1d/model.mlir +++ b/iree_tests/onnx/node/generated/test_cumsum_1d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cumsum_1d(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cumsum_1d(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.CumSum"(%arg0, %arg1) : (!torch.vtensor<[5],f64>, !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> return %0 : !torch.vtensor<[5],f64> diff --git a/iree_tests/onnx/node/generated/test_cumsum_1d_exclusive/model.mlir b/iree_tests/onnx/node/generated/test_cumsum_1d_exclusive/model.mlir index 8a0552cb4..ef99d056c 100644 --- a/iree_tests/onnx/node/generated/test_cumsum_1d_exclusive/model.mlir +++ b/iree_tests/onnx/node/generated/test_cumsum_1d_exclusive/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cumsum_1d_exclusive(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cumsum_1d_exclusive(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.CumSum"(%arg0, %arg1) {torch.onnx.exclusive = 1 : si64} : (!torch.vtensor<[5],f64>, !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> return %0 : !torch.vtensor<[5],f64> diff --git a/iree_tests/onnx/node/generated/test_cumsum_1d_reverse/model.mlir b/iree_tests/onnx/node/generated/test_cumsum_1d_reverse/model.mlir index e7bafcb02..53a9a6765 100644 --- a/iree_tests/onnx/node/generated/test_cumsum_1d_reverse/model.mlir +++ b/iree_tests/onnx/node/generated/test_cumsum_1d_reverse/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cumsum_1d_reverse(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cumsum_1d_reverse(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.CumSum"(%arg0, %arg1) {torch.onnx.reverse = 1 : si64} : (!torch.vtensor<[5],f64>, !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> return %0 : !torch.vtensor<[5],f64> diff --git a/iree_tests/onnx/node/generated/test_cumsum_1d_reverse_exclusive/model.mlir b/iree_tests/onnx/node/generated/test_cumsum_1d_reverse_exclusive/model.mlir index 69b9b4148..42b9afde0 100644 --- a/iree_tests/onnx/node/generated/test_cumsum_1d_reverse_exclusive/model.mlir +++ b/iree_tests/onnx/node/generated/test_cumsum_1d_reverse_exclusive/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cumsum_1d_reverse_exclusive(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cumsum_1d_reverse_exclusive(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.CumSum"(%arg0, %arg1) {torch.onnx.exclusive = 1 : si64, torch.onnx.reverse = 1 : si64} : (!torch.vtensor<[5],f64>, !torch.vtensor<[],si32>) -> !torch.vtensor<[5],f64> return %0 : !torch.vtensor<[5],f64> diff --git a/iree_tests/onnx/node/generated/test_cumsum_2d_axis_0/model.mlir b/iree_tests/onnx/node/generated/test_cumsum_2d_axis_0/model.mlir index c670ee02b..6b5101a11 100644 --- a/iree_tests/onnx/node/generated/test_cumsum_2d_axis_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_cumsum_2d_axis_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cumsum_2d_axis_0(%arg0: !torch.vtensor<[2,3],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cumsum_2d_axis_0(%arg0: !torch.vtensor<[2,3],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.CumSum"(%arg0, %arg1) : (!torch.vtensor<[2,3],f64>, !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> return %0 : !torch.vtensor<[2,3],f64> diff --git a/iree_tests/onnx/node/generated/test_cumsum_2d_axis_1/model.mlir b/iree_tests/onnx/node/generated/test_cumsum_2d_axis_1/model.mlir index 3ede05bf7..fbf25a48a 100644 --- a/iree_tests/onnx/node/generated/test_cumsum_2d_axis_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_cumsum_2d_axis_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cumsum_2d_axis_1(%arg0: !torch.vtensor<[2,3],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cumsum_2d_axis_1(%arg0: !torch.vtensor<[2,3],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.CumSum"(%arg0, %arg1) : (!torch.vtensor<[2,3],f64>, !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> return %0 : !torch.vtensor<[2,3],f64> diff --git a/iree_tests/onnx/node/generated/test_cumsum_2d_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_cumsum_2d_negative_axis/model.mlir index bc8703bee..bc33761bd 100644 --- a/iree_tests/onnx/node/generated/test_cumsum_2d_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_cumsum_2d_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_cumsum_2d_negative_axis(%arg0: !torch.vtensor<[2,3],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_cumsum_2d_negative_axis(%arg0: !torch.vtensor<[2,3],f64>, %arg1: !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.CumSum"(%arg0, %arg1) : (!torch.vtensor<[2,3],f64>, !torch.vtensor<[],si32>) -> !torch.vtensor<[2,3],f64> return %0 : !torch.vtensor<[2,3],f64> diff --git a/iree_tests/onnx/node/generated/test_depthtospace_crd_mode_example/model.mlir b/iree_tests/onnx/node/generated/test_depthtospace_crd_mode_example/model.mlir index 25434b483..2478db533 100644 --- a/iree_tests/onnx/node/generated/test_depthtospace_crd_mode_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_depthtospace_crd_mode_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_depthtospace_crd_mode_example(%arg0: !torch.vtensor<[1,8,2,3],f32>) -> !torch.vtensor<[1,2,4,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_depthtospace_crd_mode_example(%arg0: !torch.vtensor<[1,8,2,3],f32>) -> !torch.vtensor<[1,2,4,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.DepthToSpace"(%arg0) {torch.onnx.blocksize = 2 : si64, torch.onnx.mode = "CRD"} : (!torch.vtensor<[1,8,2,3],f32>) -> !torch.vtensor<[1,2,4,6],f32> return %0 : !torch.vtensor<[1,2,4,6],f32> diff --git a/iree_tests/onnx/node/generated/test_depthtospace_example/model.mlir b/iree_tests/onnx/node/generated/test_depthtospace_example/model.mlir index 9619288ac..f9c9bcebf 100644 --- a/iree_tests/onnx/node/generated/test_depthtospace_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_depthtospace_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_depthtospace_example(%arg0: !torch.vtensor<[1,8,2,3],f32>) -> !torch.vtensor<[1,2,4,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_depthtospace_example(%arg0: !torch.vtensor<[1,8,2,3],f32>) -> !torch.vtensor<[1,2,4,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.DepthToSpace"(%arg0) {torch.onnx.blocksize = 2 : si64, torch.onnx.mode = "DCR"} : (!torch.vtensor<[1,8,2,3],f32>) -> !torch.vtensor<[1,2,4,6],f32> return %0 : !torch.vtensor<[1,2,4,6],f32> diff --git a/iree_tests/onnx/node/generated/test_det_2d/model.mlir b/iree_tests/onnx/node/generated/test_det_2d/model.mlir index 74a4279c9..5850d053c 100644 --- a/iree_tests/onnx/node/generated/test_det_2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_det_2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_det_2d(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_det_2d(%arg0: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Det"(%arg0) : (!torch.vtensor<[2,2],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_det_nd/model.mlir b/iree_tests/onnx/node/generated/test_det_nd/model.mlir index 8b555f9fc..a0bc2b58a 100644 --- a/iree_tests/onnx/node/generated/test_det_nd/model.mlir +++ b/iree_tests/onnx/node/generated/test_det_nd/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_det_nd(%arg0: !torch.vtensor<[3,2,2],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_det_nd(%arg0: !torch.vtensor<[3,2,2],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Det"(%arg0) : (!torch.vtensor<[3,2,2],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_div/model.mlir b/iree_tests/onnx/node/generated/test_div/model.mlir index fc61a9e31..ca9288c12 100644 --- a/iree_tests/onnx/node/generated/test_div/model.mlir +++ b/iree_tests/onnx/node/generated/test_div/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_div(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_div(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Div"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_div_bcast/model.mlir b/iree_tests/onnx/node/generated/test_div_bcast/model.mlir index a809ffefc..cfd23998e 100644 --- a/iree_tests/onnx/node/generated/test_div_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_div_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_div_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_div_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Div"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_div_example/model.mlir b/iree_tests/onnx/node/generated/test_div_example/model.mlir index 9e663a913..31e08511b 100644 --- a/iree_tests/onnx/node/generated/test_div_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_div_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_div_example(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_div_example(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Div"(%arg0, %arg1) : (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> return %0 : !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_div_uint8/model.mlir b/iree_tests/onnx/node/generated/test_div_uint8/model.mlir index 28a6b0510..c4346895f 100644 --- a/iree_tests/onnx/node/generated/test_div_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_div_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_div_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_div_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Div"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],ui8>, !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> return %0 : !torch.vtensor<[3,4,5],ui8> diff --git a/iree_tests/onnx/node/generated/test_dropout_default/model.mlir b/iree_tests/onnx/node/generated/test_dropout_default/model.mlir index 7fb5684dc..7dffadd81 100644 --- a/iree_tests/onnx/node/generated/test_dropout_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_dropout_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dropout_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dropout_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Dropout"(%arg0) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_dropout_default_mask/model.mlir b/iree_tests/onnx/node/generated/test_dropout_default_mask/model.mlir index 4122aab71..973780f62 100644 --- a/iree_tests/onnx/node/generated/test_dropout_default_mask/model.mlir +++ b/iree_tests/onnx/node/generated/test_dropout_default_mask/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dropout_default_mask(%arg0: !torch.vtensor<[3,4,5],f32>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dropout_default_mask(%arg0: !torch.vtensor<[3,4,5],f32>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Dropout"(%arg0) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) return %0#0, %0#1 : !torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_dropout_default_mask_ratio/model.mlir b/iree_tests/onnx/node/generated/test_dropout_default_mask_ratio/model.mlir index d91db1374..40a158767 100644 --- a/iree_tests/onnx/node/generated/test_dropout_default_mask_ratio/model.mlir +++ b/iree_tests/onnx/node/generated/test_dropout_default_mask_ratio/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dropout_default_mask_ratio(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dropout_default_mask_ratio(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Dropout"(%arg0, %arg1) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) return %0#0, %0#1 : !torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_dropout_default_old/model.mlir b/iree_tests/onnx/node/generated/test_dropout_default_old/model.mlir index 69c4019d0..9740c30cf 100644 --- a/iree_tests/onnx/node/generated/test_dropout_default_old/model.mlir +++ b/iree_tests/onnx/node/generated/test_dropout_default_old/model.mlir @@ -1,8 +1,9 @@ module { - func.func @test_dropout_default_old(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dropout_default_old(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none - %0 = torch.operator "onnx.Dropout"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> - return %0 : !torch.vtensor<[3],f32> + %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<5.000000e-01> : tensor} : () -> !torch.vtensor<[],f32> + %1 = torch.operator "onnx.Dropout"(%arg0, %0) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> + return %1 : !torch.vtensor<[3],f32> } } diff --git a/iree_tests/onnx/node/generated/test_dropout_default_ratio/model.mlir b/iree_tests/onnx/node/generated/test_dropout_default_ratio/model.mlir index 11b4ac3a4..4924e96c8 100644 --- a/iree_tests/onnx/node/generated/test_dropout_default_ratio/model.mlir +++ b/iree_tests/onnx/node/generated/test_dropout_default_ratio/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dropout_default_ratio(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dropout_default_ratio(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Dropout"(%arg0, %arg1) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_dropout_random_old/model.mlir b/iree_tests/onnx/node/generated/test_dropout_random_old/model.mlir index a70199380..f86eecf91 100644 --- a/iree_tests/onnx/node/generated/test_dropout_random_old/model.mlir +++ b/iree_tests/onnx/node/generated/test_dropout_random_old/model.mlir @@ -1,8 +1,9 @@ module { - func.func @test_dropout_random_old(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dropout_random_old(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none - %0 = torch.operator "onnx.Dropout"(%arg0) {torch.onnx.ratio = 2.000000e-01 : f32} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> - return %0 : !torch.vtensor<[3,4,5],f32> + %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2.000000e-01> : tensor} : () -> !torch.vtensor<[],f32> + %1 = torch.operator "onnx.Dropout"(%arg0, %0) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3,4,5],f32> + return %1 : !torch.vtensor<[3,4,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_dynamicquantizelinear/model.mlir b/iree_tests/onnx/node/generated/test_dynamicquantizelinear/model.mlir index 5bac5e14f..f6da3465f 100644 --- a/iree_tests/onnx/node/generated/test_dynamicquantizelinear/model.mlir +++ b/iree_tests/onnx/node/generated/test_dynamicquantizelinear/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dynamicquantizelinear(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dynamicquantizelinear(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.DynamicQuantizeLinear"(%arg0) : (!torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) return %0#0, %0#1, %0#2 : !torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8> diff --git a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_expanded/model.mlir b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_expanded/model.mlir index 93053fc65..45b4150e6 100644 --- a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dynamicquantizelinear_expanded(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dynamicquantizelinear_expanded(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0.000000e+00> : tensor} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2.550000e+02> : tensor} : () -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted/model.mlir b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted/model.mlir index 24856c272..62ff46c32 100644 --- a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted/model.mlir +++ b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dynamicquantizelinear_max_adjusted(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dynamicquantizelinear_max_adjusted(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.DynamicQuantizeLinear"(%arg0) : (!torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) return %0#0, %0#1, %0#2 : !torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8> diff --git a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted_expanded/model.mlir b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted_expanded/model.mlir index d2c07d823..9a0581a61 100644 --- a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_max_adjusted_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dynamicquantizelinear_max_adjusted_expanded(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dynamicquantizelinear_max_adjusted_expanded(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[6],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0.000000e+00> : tensor} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2.550000e+02> : tensor} : () -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted/model.mlir b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted/model.mlir index a2d07627e..193d27aae 100644 --- a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted/model.mlir +++ b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dynamicquantizelinear_min_adjusted(%arg0: !torch.vtensor<[3,4],f32>) -> (!torch.vtensor<[3,4],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dynamicquantizelinear_min_adjusted(%arg0: !torch.vtensor<[3,4],f32>) -> (!torch.vtensor<[3,4],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.DynamicQuantizeLinear"(%arg0) : (!torch.vtensor<[3,4],f32>) -> (!torch.vtensor<[3,4],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) return %0#0, %0#1, %0#2 : !torch.vtensor<[3,4],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8> diff --git a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted_expanded/model.mlir b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted_expanded/model.mlir index 2798b5df9..22f547a25 100644 --- a/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_dynamicquantizelinear_min_adjusted_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_dynamicquantizelinear_min_adjusted_expanded(%arg0: !torch.vtensor<[3,4],f32>) -> (!torch.vtensor<[3,4],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_dynamicquantizelinear_min_adjusted_expanded(%arg0: !torch.vtensor<[3,4],f32>) -> (!torch.vtensor<[3,4],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0.000000e+00> : tensor} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2.550000e+02> : tensor} : () -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_einsum_batch_diagonal/model.mlir b/iree_tests/onnx/node/generated/test_einsum_batch_diagonal/model.mlir index 8d87726ed..59206c9fc 100644 --- a/iree_tests/onnx/node/generated/test_einsum_batch_diagonal/model.mlir +++ b/iree_tests/onnx/node/generated/test_einsum_batch_diagonal/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_einsum_batch_diagonal(%arg0: !torch.vtensor<[3,5,5],f64>) -> !torch.vtensor<[3,5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_einsum_batch_diagonal(%arg0: !torch.vtensor<[3,5,5],f64>) -> !torch.vtensor<[3,5],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Einsum"(%arg0) {torch.onnx.equation = "...ii ->...i"} : (!torch.vtensor<[3,5,5],f64>) -> !torch.vtensor<[3,5],f64> return %0 : !torch.vtensor<[3,5],f64> diff --git a/iree_tests/onnx/node/generated/test_einsum_batch_matmul/model.mlir b/iree_tests/onnx/node/generated/test_einsum_batch_matmul/model.mlir index 72c507286..e1eabb6ec 100644 --- a/iree_tests/onnx/node/generated/test_einsum_batch_matmul/model.mlir +++ b/iree_tests/onnx/node/generated/test_einsum_batch_matmul/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_einsum_batch_matmul(%arg0: !torch.vtensor<[5,2,3],f64>, %arg1: !torch.vtensor<[5,3,4],f64>) -> !torch.vtensor<[5,2,4],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_einsum_batch_matmul(%arg0: !torch.vtensor<[5,2,3],f64>, %arg1: !torch.vtensor<[5,3,4],f64>) -> !torch.vtensor<[5,2,4],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Einsum"(%arg0, %arg1) {torch.onnx.equation = "bij, bjk -> bik"} : (!torch.vtensor<[5,2,3],f64>, !torch.vtensor<[5,3,4],f64>) -> !torch.vtensor<[5,2,4],f64> return %0 : !torch.vtensor<[5,2,4],f64> diff --git a/iree_tests/onnx/node/generated/test_einsum_inner_prod/model.mlir b/iree_tests/onnx/node/generated/test_einsum_inner_prod/model.mlir index 91881e69e..4a827058a 100644 --- a/iree_tests/onnx/node/generated/test_einsum_inner_prod/model.mlir +++ b/iree_tests/onnx/node/generated/test_einsum_inner_prod/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_einsum_inner_prod(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[5],f64>) -> !torch.vtensor<[],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_einsum_inner_prod(%arg0: !torch.vtensor<[5],f64>, %arg1: !torch.vtensor<[5],f64>) -> !torch.vtensor<[],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Einsum"(%arg0, %arg1) {torch.onnx.equation = "i,i"} : (!torch.vtensor<[5],f64>, !torch.vtensor<[5],f64>) -> !torch.vtensor<[],f64> return %0 : !torch.vtensor<[],f64> diff --git a/iree_tests/onnx/node/generated/test_einsum_sum/model.mlir b/iree_tests/onnx/node/generated/test_einsum_sum/model.mlir index 4e86092e5..038d92540 100644 --- a/iree_tests/onnx/node/generated/test_einsum_sum/model.mlir +++ b/iree_tests/onnx/node/generated/test_einsum_sum/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_einsum_sum(%arg0: !torch.vtensor<[3,4],f64>) -> !torch.vtensor<[3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_einsum_sum(%arg0: !torch.vtensor<[3,4],f64>) -> !torch.vtensor<[3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Einsum"(%arg0) {torch.onnx.equation = "ij->i"} : (!torch.vtensor<[3,4],f64>) -> !torch.vtensor<[3],f64> return %0 : !torch.vtensor<[3],f64> diff --git a/iree_tests/onnx/node/generated/test_einsum_transpose/model.mlir b/iree_tests/onnx/node/generated/test_einsum_transpose/model.mlir index 8a3059b21..f46e66887 100644 --- a/iree_tests/onnx/node/generated/test_einsum_transpose/model.mlir +++ b/iree_tests/onnx/node/generated/test_einsum_transpose/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_einsum_transpose(%arg0: !torch.vtensor<[3,4],f64>) -> !torch.vtensor<[4,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_einsum_transpose(%arg0: !torch.vtensor<[3,4],f64>) -> !torch.vtensor<[4,3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Einsum"(%arg0) {torch.onnx.equation = "ij->ji"} : (!torch.vtensor<[3,4],f64>) -> !torch.vtensor<[4,3],f64> return %0 : !torch.vtensor<[4,3],f64> diff --git a/iree_tests/onnx/node/generated/test_elu/model.mlir b/iree_tests/onnx/node/generated/test_elu/model.mlir index b69f64eb7..4a60fc1ae 100644 --- a/iree_tests/onnx/node/generated/test_elu/model.mlir +++ b/iree_tests/onnx/node/generated/test_elu/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_elu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_elu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Elu"(%arg0) {torch.onnx.alpha = 2.000000e+00 : f32} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_elu_default/model.mlir b/iree_tests/onnx/node/generated/test_elu_default/model.mlir index 9ef9fa26b..2be7c703b 100644 --- a/iree_tests/onnx/node/generated/test_elu_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_elu_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_elu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_elu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Elu"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_elu_example/model.mlir b/iree_tests/onnx/node/generated/test_elu_example/model.mlir index b71e2cff7..1e8b208ef 100644 --- a/iree_tests/onnx/node/generated/test_elu_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_elu_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_elu_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_elu_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Elu"(%arg0) {torch.onnx.alpha = 2.000000e+00 : f32} : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_erf/model.mlir b/iree_tests/onnx/node/generated/test_erf/model.mlir index 5d33db016..d1c22ec39 100644 --- a/iree_tests/onnx/node/generated/test_erf/model.mlir +++ b/iree_tests/onnx/node/generated/test_erf/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_erf(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_erf(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Erf"(%arg0) : (!torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> return %0 : !torch.vtensor<[1,3,32,32],f32> diff --git a/iree_tests/onnx/node/generated/test_exp/model.mlir b/iree_tests/onnx/node/generated/test_exp/model.mlir index 4f7446aa8..1d4d6c69c 100644 --- a/iree_tests/onnx/node/generated/test_exp/model.mlir +++ b/iree_tests/onnx/node/generated/test_exp/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_exp(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_exp(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Exp"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_exp_example/model.mlir b/iree_tests/onnx/node/generated/test_exp_example/model.mlir index d93ef0d4f..074556e91 100644 --- a/iree_tests/onnx/node/generated/test_exp_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_exp_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_exp_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_exp_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Exp"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_expand_dim_changed/model.mlir b/iree_tests/onnx/node/generated/test_expand_dim_changed/model.mlir index e0e4f35a7..2212369b9 100644 --- a/iree_tests/onnx/node/generated/test_expand_dim_changed/model.mlir +++ b/iree_tests/onnx/node/generated/test_expand_dim_changed/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_expand_dim_changed(%arg0: !torch.vtensor<[3,1],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[2,3,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_expand_dim_changed(%arg0: !torch.vtensor<[3,1],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[2,3,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Expand"(%arg0, %arg1) : (!torch.vtensor<[3,1],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[2,3,6],f32> return %0 : !torch.vtensor<[2,3,6],f32> diff --git a/iree_tests/onnx/node/generated/test_expand_dim_unchanged/model.mlir b/iree_tests/onnx/node/generated/test_expand_dim_unchanged/model.mlir index da78a0564..e4a4d9d75 100644 --- a/iree_tests/onnx/node/generated/test_expand_dim_unchanged/model.mlir +++ b/iree_tests/onnx/node/generated/test_expand_dim_unchanged/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_expand_dim_unchanged(%arg0: !torch.vtensor<[3,1],f32>, %arg1: !torch.vtensor<[2],si64>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_expand_dim_unchanged(%arg0: !torch.vtensor<[3,1],f32>, %arg1: !torch.vtensor<[2],si64>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Expand"(%arg0, %arg1) : (!torch.vtensor<[3,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[3,4],f32> return %0 : !torch.vtensor<[3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_eyelike_populate_off_main_diagonal/model.mlir b/iree_tests/onnx/node/generated/test_eyelike_populate_off_main_diagonal/model.mlir index c6a5caf35..e4c4291c0 100644 --- a/iree_tests/onnx/node/generated/test_eyelike_populate_off_main_diagonal/model.mlir +++ b/iree_tests/onnx/node/generated/test_eyelike_populate_off_main_diagonal/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_eyelike_populate_off_main_diagonal(%arg0: !torch.vtensor<[4,5],si32>) -> !torch.vtensor<[4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_eyelike_populate_off_main_diagonal(%arg0: !torch.vtensor<[4,5],si32>) -> !torch.vtensor<[4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.EyeLike"(%arg0) {torch.onnx.dtype = 1 : si64, torch.onnx.k = 1 : si64} : (!torch.vtensor<[4,5],si32>) -> !torch.vtensor<[4,5],f32> return %0 : !torch.vtensor<[4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_eyelike_with_dtype/model.mlir b/iree_tests/onnx/node/generated/test_eyelike_with_dtype/model.mlir index 441ca94b3..8165023c5 100644 --- a/iree_tests/onnx/node/generated/test_eyelike_with_dtype/model.mlir +++ b/iree_tests/onnx/node/generated/test_eyelike_with_dtype/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_eyelike_with_dtype(%arg0: !torch.vtensor<[3,4],si32>) -> !torch.vtensor<[3,4],f64> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_eyelike_with_dtype(%arg0: !torch.vtensor<[3,4],si32>) -> !torch.vtensor<[3,4],f64> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.EyeLike"(%arg0) {torch.onnx.dtype = 11 : si64} : (!torch.vtensor<[3,4],si32>) -> !torch.vtensor<[3,4],f64> return %0 : !torch.vtensor<[3,4],f64> diff --git a/iree_tests/onnx/node/generated/test_eyelike_without_dtype/model.mlir b/iree_tests/onnx/node/generated/test_eyelike_without_dtype/model.mlir index 50e76ca88..cb8152612 100644 --- a/iree_tests/onnx/node/generated/test_eyelike_without_dtype/model.mlir +++ b/iree_tests/onnx/node/generated/test_eyelike_without_dtype/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_eyelike_without_dtype(%arg0: !torch.vtensor<[4,4],si32>) -> !torch.vtensor<[4,4],si32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_eyelike_without_dtype(%arg0: !torch.vtensor<[4,4],si32>) -> !torch.vtensor<[4,4],si32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.EyeLike"(%arg0) : (!torch.vtensor<[4,4],si32>) -> !torch.vtensor<[4,4],si32> return %0 : !torch.vtensor<[4,4],si32> diff --git a/iree_tests/onnx/node/generated/test_floor/model.mlir b/iree_tests/onnx/node/generated/test_floor/model.mlir index edf07030a..dbb45e8e6 100644 --- a/iree_tests/onnx/node/generated/test_floor/model.mlir +++ b/iree_tests/onnx/node/generated/test_floor/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_floor(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_floor(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Floor"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_floor_example/model.mlir b/iree_tests/onnx/node/generated/test_floor_example/model.mlir index 28a0c1a8f..a73431ee3 100644 --- a/iree_tests/onnx/node/generated/test_floor_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_floor_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_floor_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_floor_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Floor"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_gather_0/model.mlir b/iree_tests/onnx/node/generated/test_gather_0/model.mlir index 330c10595..bac72df2b 100644 --- a/iree_tests/onnx/node/generated/test_gather_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_gather_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gather_0(%arg0: !torch.vtensor<[5,4,3,2],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,4,3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gather_0(%arg0: !torch.vtensor<[5,4,3,2],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,4,3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gather"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[5,4,3,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,4,3,2],f32> return %0 : !torch.vtensor<[3,4,3,2],f32> diff --git a/iree_tests/onnx/node/generated/test_gather_1/model.mlir b/iree_tests/onnx/node/generated/test_gather_1/model.mlir index cbe9e0c33..3755ec01d 100644 --- a/iree_tests/onnx/node/generated/test_gather_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_gather_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gather_1(%arg0: !torch.vtensor<[5,4,3,2],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[5,3,3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gather_1(%arg0: !torch.vtensor<[5,4,3,2],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[5,3,3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gather"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[5,4,3,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[5,3,3,2],f32> return %0 : !torch.vtensor<[5,3,3,2],f32> diff --git a/iree_tests/onnx/node/generated/test_gather_2d_indices/model.mlir b/iree_tests/onnx/node/generated/test_gather_2d_indices/model.mlir index dc4526565..550af893d 100644 --- a/iree_tests/onnx/node/generated/test_gather_2d_indices/model.mlir +++ b/iree_tests/onnx/node/generated/test_gather_2d_indices/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gather_2d_indices(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[1,2],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gather_2d_indices(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[1,2],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gather"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[3,3],f32>, !torch.vtensor<[1,2],si64>) -> !torch.vtensor<[3,1,2],f32> return %0 : !torch.vtensor<[3,1,2],f32> diff --git a/iree_tests/onnx/node/generated/test_gather_elements_0/model.mlir b/iree_tests/onnx/node/generated/test_gather_elements_0/model.mlir index 5242133b6..ec7b5a52b 100644 --- a/iree_tests/onnx/node/generated/test_gather_elements_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_gather_elements_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gather_elements_0(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gather_elements_0(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GatherElements"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2,2],f32> return %0 : !torch.vtensor<[2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_gather_elements_1/model.mlir b/iree_tests/onnx/node/generated/test_gather_elements_1/model.mlir index c372abdc0..b0cf0da88 100644 --- a/iree_tests/onnx/node/generated/test_gather_elements_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_gather_elements_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gather_elements_1(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[2,3],si64>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gather_elements_1(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[2,3],si64>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GatherElements"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[3,3],f32>, !torch.vtensor<[2,3],si64>) -> !torch.vtensor<[2,3],f32> return %0 : !torch.vtensor<[2,3],f32> diff --git a/iree_tests/onnx/node/generated/test_gather_elements_negative_indices/model.mlir b/iree_tests/onnx/node/generated/test_gather_elements_negative_indices/model.mlir index 290e0db95..9a01b25d8 100644 --- a/iree_tests/onnx/node/generated/test_gather_elements_negative_indices/model.mlir +++ b/iree_tests/onnx/node/generated/test_gather_elements_negative_indices/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gather_elements_negative_indices(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[2,3],si64>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gather_elements_negative_indices(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[2,3],si64>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GatherElements"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[3,3],f32>, !torch.vtensor<[2,3],si64>) -> !torch.vtensor<[2,3],f32> return %0 : !torch.vtensor<[2,3],f32> diff --git a/iree_tests/onnx/node/generated/test_gather_negative_indices/model.mlir b/iree_tests/onnx/node/generated/test_gather_negative_indices/model.mlir index 6f7b60f07..906b5e1b8 100644 --- a/iree_tests/onnx/node/generated/test_gather_negative_indices/model.mlir +++ b/iree_tests/onnx/node/generated/test_gather_negative_indices/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gather_negative_indices(%arg0: !torch.vtensor<[10],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gather_negative_indices(%arg0: !torch.vtensor<[10],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gather"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[10],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_gathernd_example_float32/model.mlir b/iree_tests/onnx/node/generated/test_gathernd_example_float32/model.mlir index 03b1188ac..8d26c0481 100644 --- a/iree_tests/onnx/node/generated/test_gathernd_example_float32/model.mlir +++ b/iree_tests/onnx/node/generated/test_gathernd_example_float32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gathernd_example_float32(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,1,2],si64>) -> !torch.vtensor<[2,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gathernd_example_float32(%arg0: !torch.vtensor<[2,2,2],f32>, %arg1: !torch.vtensor<[2,1,2],si64>) -> !torch.vtensor<[2,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GatherND"(%arg0, %arg1) : (!torch.vtensor<[2,2,2],f32>, !torch.vtensor<[2,1,2],si64>) -> !torch.vtensor<[2,1,2],f32> return %0 : !torch.vtensor<[2,1,2],f32> diff --git a/iree_tests/onnx/node/generated/test_gathernd_example_int32/model.mlir b/iree_tests/onnx/node/generated/test_gathernd_example_int32/model.mlir index f6e3ca6a8..b4ef2b7c9 100644 --- a/iree_tests/onnx/node/generated/test_gathernd_example_int32/model.mlir +++ b/iree_tests/onnx/node/generated/test_gathernd_example_int32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gathernd_example_int32(%arg0: !torch.vtensor<[2,2],si32>, %arg1: !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gathernd_example_int32(%arg0: !torch.vtensor<[2,2],si32>, %arg1: !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GatherND"(%arg0, %arg1) : (!torch.vtensor<[2,2],si32>, !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2],si32> return %0 : !torch.vtensor<[2],si32> diff --git a/iree_tests/onnx/node/generated/test_gathernd_example_int32_batch_dim1/model.mlir b/iree_tests/onnx/node/generated/test_gathernd_example_int32_batch_dim1/model.mlir index 09b2f3560..33829b85d 100644 --- a/iree_tests/onnx/node/generated/test_gathernd_example_int32_batch_dim1/model.mlir +++ b/iree_tests/onnx/node/generated/test_gathernd_example_int32_batch_dim1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gathernd_example_int32_batch_dim1(%arg0: !torch.vtensor<[2,2,2],si32>, %arg1: !torch.vtensor<[2,1],si64>) -> !torch.vtensor<[2,2],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gathernd_example_int32_batch_dim1(%arg0: !torch.vtensor<[2,2,2],si32>, %arg1: !torch.vtensor<[2,1],si64>) -> !torch.vtensor<[2,2],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GatherND"(%arg0, %arg1) {torch.onnx.batch_dims = 1 : si64} : (!torch.vtensor<[2,2,2],si32>, !torch.vtensor<[2,1],si64>) -> !torch.vtensor<[2,2],si32> return %0 : !torch.vtensor<[2,2],si32> diff --git a/iree_tests/onnx/node/generated/test_gemm_all_attributes/model.mlir b/iree_tests/onnx/node/generated/test_gemm_all_attributes/model.mlir index 3ade5bc0b..17d988d96 100644 --- a/iree_tests/onnx/node/generated/test_gemm_all_attributes/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_all_attributes/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_all_attributes(%arg0: !torch.vtensor<[4,3],f32>, %arg1: !torch.vtensor<[5,4],f32>, %arg2: !torch.vtensor<[1,5],f32>) -> !torch.vtensor<[3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_all_attributes(%arg0: !torch.vtensor<[4,3],f32>, %arg1: !torch.vtensor<[5,4],f32>, %arg2: !torch.vtensor<[1,5],f32>) -> !torch.vtensor<[3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) {torch.onnx.alpha = 2.500000e-01 : f32, torch.onnx.beta = 3.500000e-01 : f32, torch.onnx.transA = 1 : si64, torch.onnx.transB = 1 : si64} : (!torch.vtensor<[4,3],f32>, !torch.vtensor<[5,4],f32>, !torch.vtensor<[1,5],f32>) -> !torch.vtensor<[3,5],f32> return %0 : !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_alpha/model.mlir b/iree_tests/onnx/node/generated/test_gemm_alpha/model.mlir index 59c0bdafe..ebfc9953e 100644 --- a/iree_tests/onnx/node/generated/test_gemm_alpha/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_alpha/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_alpha(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[5,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_alpha(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[5,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) {torch.onnx.alpha = 5.000000e-01 : f32} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[5,4],f32>, !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> return %0 : !torch.vtensor<[3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_beta/model.mlir b/iree_tests/onnx/node/generated/test_gemm_beta/model.mlir index 8174c7471..eb315334e 100644 --- a/iree_tests/onnx/node/generated/test_gemm_beta/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_beta/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_beta(%arg0: !torch.vtensor<[2,7],f32>, %arg1: !torch.vtensor<[7,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_beta(%arg0: !torch.vtensor<[2,7],f32>, %arg1: !torch.vtensor<[7,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) {torch.onnx.beta = 5.000000e-01 : f32} : (!torch.vtensor<[2,7],f32>, !torch.vtensor<[7,4],f32>, !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[2,4],f32> return %0 : !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_default_matrix_bias/model.mlir b/iree_tests/onnx/node/generated/test_gemm_default_matrix_bias/model.mlir index 145952791..c7668f03a 100644 --- a/iree_tests/onnx/node/generated/test_gemm_default_matrix_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_default_matrix_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_default_matrix_bias(%arg0: !torch.vtensor<[3,6],f32>, %arg1: !torch.vtensor<[6,4],f32>, %arg2: !torch.vtensor<[3,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_default_matrix_bias(%arg0: !torch.vtensor<[3,6],f32>, %arg1: !torch.vtensor<[6,4],f32>, %arg2: !torch.vtensor<[3,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3,6],f32>, !torch.vtensor<[6,4],f32>, !torch.vtensor<[3,4],f32>) -> !torch.vtensor<[3,4],f32> return %0 : !torch.vtensor<[3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_default_no_bias/model.mlir b/iree_tests/onnx/node/generated/test_gemm_default_no_bias/model.mlir index 6ea6e6f2e..1bc4bfdf9 100644 --- a/iree_tests/onnx/node/generated/test_gemm_default_no_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_default_no_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_default_no_bias(%arg0: !torch.vtensor<[2,10],f32>, %arg1: !torch.vtensor<[10,3],f32>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_default_no_bias(%arg0: !torch.vtensor<[2,10],f32>, %arg1: !torch.vtensor<[10,3],f32>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1) : (!torch.vtensor<[2,10],f32>, !torch.vtensor<[10,3],f32>) -> !torch.vtensor<[2,3],f32> return %0 : !torch.vtensor<[2,3],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_default_scalar_bias/model.mlir b/iree_tests/onnx/node/generated/test_gemm_default_scalar_bias/model.mlir index 29a0280c2..8f2351290 100644 --- a/iree_tests/onnx/node/generated/test_gemm_default_scalar_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_default_scalar_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_default_scalar_bias(%arg0: !torch.vtensor<[2,3],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_default_scalar_bias(%arg0: !torch.vtensor<[2,3],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) : (!torch.vtensor<[2,3],f32>, !torch.vtensor<[3,4],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[2,4],f32> return %0 : !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_default_single_elem_vector_bias/model.mlir b/iree_tests/onnx/node/generated/test_gemm_default_single_elem_vector_bias/model.mlir index 5a91ed965..f69f32113 100644 --- a/iree_tests/onnx/node/generated/test_gemm_default_single_elem_vector_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_default_single_elem_vector_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_default_single_elem_vector_bias(%arg0: !torch.vtensor<[3,7],f32>, %arg1: !torch.vtensor<[7,3],f32>, %arg2: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_default_single_elem_vector_bias(%arg0: !torch.vtensor<[3,7],f32>, %arg1: !torch.vtensor<[7,3],f32>, %arg2: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3,7],f32>, !torch.vtensor<[7,3],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],f32> return %0 : !torch.vtensor<[3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_default_vector_bias/model.mlir b/iree_tests/onnx/node/generated/test_gemm_default_vector_bias/model.mlir index 14ea25053..86d9a243b 100644 --- a/iree_tests/onnx/node/generated/test_gemm_default_vector_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_default_vector_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_default_vector_bias(%arg0: !torch.vtensor<[2,7],f32>, %arg1: !torch.vtensor<[7,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_default_vector_bias(%arg0: !torch.vtensor<[2,7],f32>, %arg1: !torch.vtensor<[7,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) : (!torch.vtensor<[2,7],f32>, !torch.vtensor<[7,4],f32>, !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[2,4],f32> return %0 : !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_default_zero_bias/model.mlir b/iree_tests/onnx/node/generated/test_gemm_default_zero_bias/model.mlir index 5ce77ece9..36af90fb4 100644 --- a/iree_tests/onnx/node/generated/test_gemm_default_zero_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_default_zero_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_default_zero_bias(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[5,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_default_zero_bias(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[5,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[5,4],f32>, !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> return %0 : !torch.vtensor<[3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_transposeA/model.mlir b/iree_tests/onnx/node/generated/test_gemm_transposeA/model.mlir index 798929091..bb38a2131 100644 --- a/iree_tests/onnx/node/generated/test_gemm_transposeA/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_transposeA/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_transposeA(%arg0: !torch.vtensor<[6,3],f32>, %arg1: !torch.vtensor<[6,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_transposeA(%arg0: !torch.vtensor<[6,3],f32>, %arg1: !torch.vtensor<[6,4],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) {torch.onnx.transA = 1 : si64} : (!torch.vtensor<[6,3],f32>, !torch.vtensor<[6,4],f32>, !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> return %0 : !torch.vtensor<[3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_gemm_transposeB/model.mlir b/iree_tests/onnx/node/generated/test_gemm_transposeB/model.mlir index ca3322496..a18a222fd 100644 --- a/iree_tests/onnx/node/generated/test_gemm_transposeB/model.mlir +++ b/iree_tests/onnx/node/generated/test_gemm_transposeB/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gemm_transposeB(%arg0: !torch.vtensor<[3,6],f32>, %arg1: !torch.vtensor<[4,6],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gemm_transposeB(%arg0: !torch.vtensor<[3,6],f32>, %arg1: !torch.vtensor<[4,6],f32>, %arg2: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Gemm"(%arg0, %arg1, %arg2) {torch.onnx.transB = 1 : si64} : (!torch.vtensor<[3,6],f32>, !torch.vtensor<[4,6],f32>, !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[3,4],f32> return %0 : !torch.vtensor<[3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_globalaveragepool/model.mlir b/iree_tests/onnx/node/generated/test_globalaveragepool/model.mlir index 19e70f898..22a336d97 100644 --- a/iree_tests/onnx/node/generated/test_globalaveragepool/model.mlir +++ b/iree_tests/onnx/node/generated/test_globalaveragepool/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_globalaveragepool(%arg0: !torch.vtensor<[1,3,5,5],f32>) -> !torch.vtensor<[1,3,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_globalaveragepool(%arg0: !torch.vtensor<[1,3,5,5],f32>) -> !torch.vtensor<[1,3,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GlobalAveragePool"(%arg0) : (!torch.vtensor<[1,3,5,5],f32>) -> !torch.vtensor<[1,3,1,1],f32> return %0 : !torch.vtensor<[1,3,1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_globalaveragepool_precomputed/model.mlir b/iree_tests/onnx/node/generated/test_globalaveragepool_precomputed/model.mlir index 5a5477770..979ca3f41 100644 --- a/iree_tests/onnx/node/generated/test_globalaveragepool_precomputed/model.mlir +++ b/iree_tests/onnx/node/generated/test_globalaveragepool_precomputed/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_globalaveragepool_precomputed(%arg0: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_globalaveragepool_precomputed(%arg0: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GlobalAveragePool"(%arg0) : (!torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,1,1],f32> return %0 : !torch.vtensor<[1,1,1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_globalmaxpool/model.mlir b/iree_tests/onnx/node/generated/test_globalmaxpool/model.mlir index e8e813662..2f450b7a3 100644 --- a/iree_tests/onnx/node/generated/test_globalmaxpool/model.mlir +++ b/iree_tests/onnx/node/generated/test_globalmaxpool/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_globalmaxpool(%arg0: !torch.vtensor<[1,3,5,5],f32>) -> !torch.vtensor<[1,3,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_globalmaxpool(%arg0: !torch.vtensor<[1,3,5,5],f32>) -> !torch.vtensor<[1,3,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GlobalMaxPool"(%arg0) : (!torch.vtensor<[1,3,5,5],f32>) -> !torch.vtensor<[1,3,1,1],f32> return %0 : !torch.vtensor<[1,3,1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_globalmaxpool_precomputed/model.mlir b/iree_tests/onnx/node/generated/test_globalmaxpool_precomputed/model.mlir index 84e500b02..9b0ebe6e8 100644 --- a/iree_tests/onnx/node/generated/test_globalmaxpool_precomputed/model.mlir +++ b/iree_tests/onnx/node/generated/test_globalmaxpool_precomputed/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_globalmaxpool_precomputed(%arg0: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_globalmaxpool_precomputed(%arg0: !torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,1,1],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GlobalMaxPool"(%arg0) : (!torch.vtensor<[1,1,3,3],f32>) -> !torch.vtensor<[1,1,1,1],f32> return %0 : !torch.vtensor<[1,1,1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_greater/model.mlir b/iree_tests/onnx/node/generated/test_greater/model.mlir index 83d8cdff1..84da678dc 100644 --- a/iree_tests/onnx/node/generated/test_greater/model.mlir +++ b/iree_tests/onnx/node/generated/test_greater/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_greater(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_greater(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Greater"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_greater_bcast/model.mlir b/iree_tests/onnx/node/generated/test_greater_bcast/model.mlir index 08e2a8d03..743a569e6 100644 --- a/iree_tests/onnx/node/generated/test_greater_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_greater_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_greater_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_greater_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Greater"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_greater_equal/model.mlir b/iree_tests/onnx/node/generated/test_greater_equal/model.mlir index 1b6001022..40f7e812e 100644 --- a/iree_tests/onnx/node/generated/test_greater_equal/model.mlir +++ b/iree_tests/onnx/node/generated/test_greater_equal/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_greater_equal(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_greater_equal(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GreaterOrEqual"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_greater_equal_bcast/model.mlir b/iree_tests/onnx/node/generated/test_greater_equal_bcast/model.mlir index 43ad2c9f8..32cfc267f 100644 --- a/iree_tests/onnx/node/generated/test_greater_equal_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_greater_equal_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_greater_equal_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_greater_equal_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.GreaterOrEqual"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_greater_equal_bcast_expanded/model.mlir b/iree_tests/onnx/node/generated/test_greater_equal_bcast_expanded/model.mlir index 0b6e1de76..ba31c0b11 100644 --- a/iree_tests/onnx/node/generated/test_greater_equal_bcast_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_greater_equal_bcast_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_greater_equal_bcast_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_greater_equal_bcast_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Greater"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Equal"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_greater_equal_expanded/model.mlir b/iree_tests/onnx/node/generated/test_greater_equal_expanded/model.mlir index 60ffd3273..2a5622c09 100644 --- a/iree_tests/onnx/node/generated/test_greater_equal_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_greater_equal_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_greater_equal_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_greater_equal_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Greater"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Equal"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_gru_batchwise/model.mlir b/iree_tests/onnx/node/generated/test_gru_batchwise/model.mlir index 16c483275..b0af8c043 100644 --- a/iree_tests/onnx/node/generated/test_gru_batchwise/model.mlir +++ b/iree_tests/onnx/node/generated/test_gru_batchwise/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gru_batchwise(%arg0: !torch.vtensor<[3,1,2],f32>, %arg1: !torch.vtensor<[1,18,2],f32>, %arg2: !torch.vtensor<[1,18,6],f32>) -> (!torch.vtensor<[3,1,1,6],f32>, !torch.vtensor<[3,1,6],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gru_batchwise(%arg0: !torch.vtensor<[3,1,2],f32>, %arg1: !torch.vtensor<[1,18,2],f32>, %arg2: !torch.vtensor<[1,18,6],f32>) -> (!torch.vtensor<[3,1,1,6],f32>, !torch.vtensor<[3,1,6],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.GRU"(%arg0, %arg1, %arg2) {torch.onnx.hidden_size = 6 : si64, torch.onnx.layout = 1 : si64} : (!torch.vtensor<[3,1,2],f32>, !torch.vtensor<[1,18,2],f32>, !torch.vtensor<[1,18,6],f32>) -> (!torch.vtensor<[3,1,1,6],f32>, !torch.vtensor<[3,1,6],f32>) return %0#0, %0#1 : !torch.vtensor<[3,1,1,6],f32>, !torch.vtensor<[3,1,6],f32> diff --git a/iree_tests/onnx/node/generated/test_gru_defaults/model.mlir b/iree_tests/onnx/node/generated/test_gru_defaults/model.mlir index 8f58f7305..bbaa9372a 100644 --- a/iree_tests/onnx/node/generated/test_gru_defaults/model.mlir +++ b/iree_tests/onnx/node/generated/test_gru_defaults/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gru_defaults(%arg0: !torch.vtensor<[1,3,2],f32>, %arg1: !torch.vtensor<[1,15,2],f32>, %arg2: !torch.vtensor<[1,15,5],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gru_defaults(%arg0: !torch.vtensor<[1,3,2],f32>, %arg1: !torch.vtensor<[1,15,2],f32>, %arg2: !torch.vtensor<[1,15,5],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.GRU"(%arg0, %arg1, %arg2) {torch.onnx.hidden_size = 5 : si64} : (!torch.vtensor<[1,3,2],f32>, !torch.vtensor<[1,15,2],f32>, !torch.vtensor<[1,15,5],f32>) -> (!torch.none, !torch.vtensor<[1,3,5],f32>) return %0#1 : !torch.vtensor<[1,3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_gru_seq_length/model.mlir b/iree_tests/onnx/node/generated/test_gru_seq_length/model.mlir index ba1317df4..7ca95b836 100644 --- a/iree_tests/onnx/node/generated/test_gru_seq_length/model.mlir +++ b/iree_tests/onnx/node/generated/test_gru_seq_length/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gru_seq_length(%arg0: !torch.vtensor<[2,3,3],f32>, %arg1: !torch.vtensor<[1,15,3],f32>, %arg2: !torch.vtensor<[1,15,5],f32>, %arg3: !torch.vtensor<[1,30],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gru_seq_length(%arg0: !torch.vtensor<[2,3,3],f32>, %arg1: !torch.vtensor<[1,15,3],f32>, %arg2: !torch.vtensor<[1,15,5],f32>, %arg3: !torch.vtensor<[1,30],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.GRU"(%arg0, %arg1, %arg2, %arg3) {torch.onnx.hidden_size = 5 : si64} : (!torch.vtensor<[2,3,3],f32>, !torch.vtensor<[1,15,3],f32>, !torch.vtensor<[1,15,5],f32>, !torch.vtensor<[1,30],f32>) -> (!torch.none, !torch.vtensor<[1,3,5],f32>) return %0#1 : !torch.vtensor<[1,3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_gru_with_initial_bias/model.mlir b/iree_tests/onnx/node/generated/test_gru_with_initial_bias/model.mlir index 7d67c4b46..5c6d48d24 100644 --- a/iree_tests/onnx/node/generated/test_gru_with_initial_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_gru_with_initial_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_gru_with_initial_bias(%arg0: !torch.vtensor<[1,3,3],f32>, %arg1: !torch.vtensor<[1,9,3],f32>, %arg2: !torch.vtensor<[1,9,3],f32>, %arg3: !torch.vtensor<[1,18],f32>) -> !torch.vtensor<[1,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_gru_with_initial_bias(%arg0: !torch.vtensor<[1,3,3],f32>, %arg1: !torch.vtensor<[1,9,3],f32>, %arg2: !torch.vtensor<[1,9,3],f32>, %arg3: !torch.vtensor<[1,18],f32>) -> !torch.vtensor<[1,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.GRU"(%arg0, %arg1, %arg2, %arg3) {torch.onnx.hidden_size = 3 : si64} : (!torch.vtensor<[1,3,3],f32>, !torch.vtensor<[1,9,3],f32>, !torch.vtensor<[1,9,3],f32>, !torch.vtensor<[1,18],f32>) -> (!torch.none, !torch.vtensor<[1,3,3],f32>) return %0#1 : !torch.vtensor<[1,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_hardmax_axis_0/model.mlir b/iree_tests/onnx/node/generated/test_hardmax_axis_0/model.mlir index 6661c6414..7cbafb4c1 100644 --- a/iree_tests/onnx/node/generated/test_hardmax_axis_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardmax_axis_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardmax_axis_0(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardmax_axis_0(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Hardmax"(%arg0) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardmax_axis_1/model.mlir b/iree_tests/onnx/node/generated/test_hardmax_axis_1/model.mlir index 04da38a44..949cec44f 100644 --- a/iree_tests/onnx/node/generated/test_hardmax_axis_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardmax_axis_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardmax_axis_1(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardmax_axis_1(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Hardmax"(%arg0) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardmax_axis_2/model.mlir b/iree_tests/onnx/node/generated/test_hardmax_axis_2/model.mlir index 78f789043..e171750d6 100644 --- a/iree_tests/onnx/node/generated/test_hardmax_axis_2/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardmax_axis_2/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardmax_axis_2(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardmax_axis_2(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Hardmax"(%arg0) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardmax_default_axis/model.mlir b/iree_tests/onnx/node/generated/test_hardmax_default_axis/model.mlir index 3e34df19d..5b88f1e60 100644 --- a/iree_tests/onnx/node/generated/test_hardmax_default_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardmax_default_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardmax_default_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardmax_default_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Hardmax"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardmax_example/model.mlir b/iree_tests/onnx/node/generated/test_hardmax_example/model.mlir index 294fc830c..ad311ba5b 100644 --- a/iree_tests/onnx/node/generated/test_hardmax_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardmax_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardmax_example(%arg0: !torch.vtensor<[4,4],f32>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardmax_example(%arg0: !torch.vtensor<[4,4],f32>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Hardmax"(%arg0) : (!torch.vtensor<[4,4],f32>) -> !torch.vtensor<[4,4],f32> return %0 : !torch.vtensor<[4,4],f32> diff --git a/iree_tests/onnx/node/generated/test_hardmax_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_hardmax_negative_axis/model.mlir index 620076947..bff823f86 100644 --- a/iree_tests/onnx/node/generated/test_hardmax_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardmax_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardmax_negative_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardmax_negative_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Hardmax"(%arg0) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardmax_one_hot/model.mlir b/iree_tests/onnx/node/generated/test_hardmax_one_hot/model.mlir index bd0e2cfa1..2bf125014 100644 --- a/iree_tests/onnx/node/generated/test_hardmax_one_hot/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardmax_one_hot/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardmax_one_hot(%arg0: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[1,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardmax_one_hot(%arg0: !torch.vtensor<[1,4],f32>) -> !torch.vtensor<[1,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Hardmax"(%arg0) : (!torch.vtensor<[1,4],f32>) -> !torch.vtensor<[1,4],f32> return %0 : !torch.vtensor<[1,4],f32> diff --git a/iree_tests/onnx/node/generated/test_hardsigmoid/model.mlir b/iree_tests/onnx/node/generated/test_hardsigmoid/model.mlir index 4cfce7a80..c7a02cf00 100644 --- a/iree_tests/onnx/node/generated/test_hardsigmoid/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardsigmoid/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardsigmoid(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardsigmoid(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.HardSigmoid"(%arg0) {torch.onnx.alpha = 5.000000e-01 : f32, torch.onnx.beta = 6.000000e-01 : f32} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardsigmoid_default/model.mlir b/iree_tests/onnx/node/generated/test_hardsigmoid_default/model.mlir index 0a73ebd99..481009a3a 100644 --- a/iree_tests/onnx/node/generated/test_hardsigmoid_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardsigmoid_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardsigmoid_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardsigmoid_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.HardSigmoid"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardsigmoid_example/model.mlir b/iree_tests/onnx/node/generated/test_hardsigmoid_example/model.mlir index f76e41665..b16590a4c 100644 --- a/iree_tests/onnx/node/generated/test_hardsigmoid_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardsigmoid_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardsigmoid_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardsigmoid_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.HardSigmoid"(%arg0) {torch.onnx.alpha = 5.000000e-01 : f32, torch.onnx.beta = 6.000000e-01 : f32} : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_hardswish/model.mlir b/iree_tests/onnx/node/generated/test_hardswish/model.mlir index b0e9623f9..65a740f00 100644 --- a/iree_tests/onnx/node/generated/test_hardswish/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardswish/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardswish(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardswish(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.HardSwish"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_hardswish_expanded/model.mlir b/iree_tests/onnx/node/generated/test_hardswish_expanded/model.mlir index ec7bcece2..b7b10a9d7 100644 --- a/iree_tests/onnx/node/generated/test_hardswish_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_hardswish_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_hardswish_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_hardswish_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.HardSigmoid"(%arg0) {torch.onnx.alpha = 0.166666672 : f32, torch.onnx.beta = 5.000000e-01 : f32} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> %1 = torch.operator "onnx.Mul"(%arg0, %0) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_if/model.mlir b/iree_tests/onnx/node/generated/test_if/model.mlir index 8a68479cc..488fd4cec 100644 --- a/iree_tests/onnx/node/generated/test_if/model.mlir +++ b/iree_tests/onnx/node/generated/test_if/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_if(%arg0: !torch.vtensor<[],i1>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_if(%arg0: !torch.vtensor<[],i1>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.If"(%arg0) : (!torch.vtensor<[],i1>) -> !torch.vtensor<[5],f32> { %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense_resource<_> : tensor<5xf32>} : () -> !torch.vtensor<[5],f32> diff --git a/iree_tests/onnx/node/generated/test_instancenorm_epsilon/model.mlir b/iree_tests/onnx/node/generated/test_instancenorm_epsilon/model.mlir index 1828962c7..238149c7a 100644 --- a/iree_tests/onnx/node/generated/test_instancenorm_epsilon/model.mlir +++ b/iree_tests/onnx/node/generated/test_instancenorm_epsilon/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_instancenorm_epsilon(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_instancenorm_epsilon(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.InstanceNormalization"(%arg0, %arg1, %arg2) {torch.onnx.epsilon = 0.00999999977 : f32} : (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3,4,5],f32> return %0 : !torch.vtensor<[2,3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_instancenorm_example/model.mlir b/iree_tests/onnx/node/generated/test_instancenorm_example/model.mlir index ab1a6f7a6..085d8f406 100644 --- a/iree_tests/onnx/node/generated/test_instancenorm_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_instancenorm_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_instancenorm_example(%arg0: !torch.vtensor<[1,2,1,3],f32>, %arg1: !torch.vtensor<[2],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[1,2,1,3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_instancenorm_example(%arg0: !torch.vtensor<[1,2,1,3],f32>, %arg1: !torch.vtensor<[2],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[1,2,1,3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.InstanceNormalization"(%arg0, %arg1, %arg2) : (!torch.vtensor<[1,2,1,3],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[1,2,1,3],f32> return %0 : !torch.vtensor<[1,2,1,3],f32> diff --git a/iree_tests/onnx/node/generated/test_leakyrelu/model.mlir b/iree_tests/onnx/node/generated/test_leakyrelu/model.mlir index 41fa678e5..865ba40fc 100644 --- a/iree_tests/onnx/node/generated/test_leakyrelu/model.mlir +++ b/iree_tests/onnx/node/generated/test_leakyrelu/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_leakyrelu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_leakyrelu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LeakyRelu"(%arg0) {torch.onnx.alpha = 1.000000e-01 : f32} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_leakyrelu_default/model.mlir b/iree_tests/onnx/node/generated/test_leakyrelu_default/model.mlir index b652c9942..200cba47e 100644 --- a/iree_tests/onnx/node/generated/test_leakyrelu_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_leakyrelu_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_leakyrelu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_leakyrelu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LeakyRelu"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_leakyrelu_default_expanded/model.mlir b/iree_tests/onnx/node/generated/test_leakyrelu_default_expanded/model.mlir index b388692f2..e2b08170e 100644 --- a/iree_tests/onnx/node/generated/test_leakyrelu_default_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_leakyrelu_default_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_leakyrelu_default_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_leakyrelu_default_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value_float = 0.00999999977 : f32} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.CastLike"(%0, %arg0) : (!torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_leakyrelu_example/model.mlir b/iree_tests/onnx/node/generated/test_leakyrelu_example/model.mlir index b8afdd356..5d559bc38 100644 --- a/iree_tests/onnx/node/generated/test_leakyrelu_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_leakyrelu_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_leakyrelu_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_leakyrelu_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LeakyRelu"(%arg0) {torch.onnx.alpha = 1.000000e-01 : f32} : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_leakyrelu_example_expanded/model.mlir b/iree_tests/onnx/node/generated/test_leakyrelu_example_expanded/model.mlir index 6a6cc6658..78e2792a6 100644 --- a/iree_tests/onnx/node/generated/test_leakyrelu_example_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_leakyrelu_example_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_leakyrelu_example_expanded(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_leakyrelu_example_expanded(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value_float = 1.000000e-01 : f32} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.CastLike"(%0, %arg0) : (!torch.vtensor<[],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_leakyrelu_expanded/model.mlir b/iree_tests/onnx/node/generated/test_leakyrelu_expanded/model.mlir index e53e0025e..77de395aa 100644 --- a/iree_tests/onnx/node/generated/test_leakyrelu_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_leakyrelu_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_leakyrelu_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_leakyrelu_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value_float = 1.000000e-01 : f32} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.CastLike"(%0, %arg0) : (!torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_less/model.mlir b/iree_tests/onnx/node/generated/test_less/model.mlir index 3dbf89077..b77fe3433 100644 --- a/iree_tests/onnx/node/generated/test_less/model.mlir +++ b/iree_tests/onnx/node/generated/test_less/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_less(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_less(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_less_bcast/model.mlir b/iree_tests/onnx/node/generated/test_less_bcast/model.mlir index 806f29e85..fe68c8ef0 100644 --- a/iree_tests/onnx/node/generated/test_less_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_less_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_less_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_less_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_less_equal/model.mlir b/iree_tests/onnx/node/generated/test_less_equal/model.mlir index 7a3b714ed..aa414f9dc 100644 --- a/iree_tests/onnx/node/generated/test_less_equal/model.mlir +++ b/iree_tests/onnx/node/generated/test_less_equal/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_less_equal(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_less_equal(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LessOrEqual"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_less_equal_bcast/model.mlir b/iree_tests/onnx/node/generated/test_less_equal_bcast/model.mlir index 742175ae9..ed51d39b5 100644 --- a/iree_tests/onnx/node/generated/test_less_equal_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_less_equal_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_less_equal_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_less_equal_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LessOrEqual"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_less_equal_bcast_expanded/model.mlir b/iree_tests/onnx/node/generated/test_less_equal_bcast_expanded/model.mlir index 5bdd8b3df..99caba72b 100644 --- a/iree_tests/onnx/node/generated/test_less_equal_bcast_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_less_equal_bcast_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_less_equal_bcast_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_less_equal_bcast_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Equal"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_less_equal_expanded/model.mlir b/iree_tests/onnx/node/generated/test_less_equal_expanded/model.mlir index 78b069b87..db9218ae1 100644 --- a/iree_tests/onnx/node/generated/test_less_equal_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_less_equal_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_less_equal_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_less_equal_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Less"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> %1 = torch.operator "onnx.Equal"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_log/model.mlir b/iree_tests/onnx/node/generated/test_log/model.mlir index f02616a3c..f503946c0 100644 --- a/iree_tests/onnx/node/generated/test_log/model.mlir +++ b/iree_tests/onnx/node/generated/test_log/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_log(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_log(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Log"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_log_example/model.mlir b/iree_tests/onnx/node/generated/test_log_example/model.mlir index 8e557039a..1bccb1273 100644 --- a/iree_tests/onnx/node/generated/test_log_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_log_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_log_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_log_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Log"(%arg0) : (!torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> return %0 : !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_axis_0/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_axis_0/model.mlir index 2fce0a489..de669a0b5 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_axis_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_axis_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_axis_0(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_axis_0(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LogSoftmax"(%arg0) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_axis_0_expanded/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_axis_0_expanded/model.mlir index 06515d2c9..58826bc4c 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_axis_0_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_axis_0_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_axis_0_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_axis_0_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [0 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[1,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_axis_1/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_axis_1/model.mlir index ba33a2cea..461eb8820 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_axis_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_axis_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_axis_1(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_axis_1(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LogSoftmax"(%arg0) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_axis_1_expanded/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_axis_1_expanded/model.mlir index f2ca8d8b5..2df91095f 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_axis_1_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_axis_1_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_axis_1_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_axis_1_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,1,5],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_axis_2/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_axis_2/model.mlir index 5bfe49d8a..73622bd90 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_axis_2/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_axis_2/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_axis_2(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_axis_2(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LogSoftmax"(%arg0) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_axis_2_expanded/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_axis_2_expanded/model.mlir index d8d8f0b9b..6bf8b42d8 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_axis_2_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_axis_2_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_axis_2_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_axis_2_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [2 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,1],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_default_axis/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_default_axis/model.mlir index a302da0f4..8892bcfa2 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_default_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_default_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_default_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_default_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LogSoftmax"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_default_axis_expanded/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_default_axis_expanded/model.mlir index 7a0f1730f..397538710 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_default_axis_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_default_axis_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_default_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_default_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,1],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_example_1/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_example_1/model.mlir index fd9ab81c8..5cfaf8968 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_example_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_example_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_example_1(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_example_1(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LogSoftmax"(%arg0) : (!torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> return %0 : !torch.vtensor<[1,3],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_example_1_expanded/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_example_1_expanded/model.mlir index db8d7dc1e..2263f3a83 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_example_1_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_example_1_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_example_1_expanded(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_example_1_expanded(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_large_number/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_large_number/model.mlir index 0166786a2..948059f53 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_large_number/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_large_number/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_large_number(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_large_number(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LogSoftmax"(%arg0) : (!torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> return %0 : !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_large_number_expanded/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_large_number_expanded/model.mlir index 9892f33ed..030a384c6 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_large_number_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_large_number_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_large_number_expanded(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_large_number_expanded(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,1],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis/model.mlir index fe2c57be9..3457fb7f8 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_negative_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_negative_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LogSoftmax"(%arg0) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis_expanded/model.mlir b/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis_expanded/model.mlir index 0b929b447..4511d6b69 100644 --- a/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_logsoftmax_negative_axis_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_logsoftmax_negative_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_logsoftmax_negative_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,1],f32> diff --git a/iree_tests/onnx/node/generated/test_loop11/model.mlir b/iree_tests/onnx/node/generated/test_loop11/model.mlir index 91f834771..ffd7e72a9 100644 --- a/iree_tests/onnx/node/generated/test_loop11/model.mlir +++ b/iree_tests/onnx/node/generated/test_loop11/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_loop11(%arg0: !torch.vtensor<[],si64>, %arg1: !torch.vtensor<[],i1>, %arg2: !torch.vtensor<[1],f32>) -> (!torch.vtensor<[1],f32>, !torch.vtensor<[5,1],f32>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_loop11(%arg0: !torch.vtensor<[],si64>, %arg1: !torch.vtensor<[],i1>, %arg2: !torch.vtensor<[1],f32>) -> (!torch.vtensor<[1],f32>, !torch.vtensor<[5,1],f32>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Loop"(%arg0, %arg1, %arg2) : (!torch.vtensor<[],si64>, !torch.vtensor<[],i1>, !torch.vtensor<[1],f32>) -> (!torch.vtensor<[1],f32>, !torch.vtensor<[5,1],f32>) { ^bb0(%arg3: !torch.vtensor<[],si64>, %arg4: !torch.vtensor<[],i1>, %arg5: !torch.vtensor<[1],f32>): @@ -7,12 +7,14 @@ module { %2 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[1.000000e+00, 2.000000e+00, 3.000000e+00, 4.000000e+00, 5.000000e+00]> : tensor<5xf32>} : () -> !torch.vtensor<[5],f32> %3 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor} : () -> !torch.vtensor<[],si64> %4 = torch.operator "onnx.Add"(%arg3, %3) : (!torch.vtensor<[],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[],si64> - %5 = torch.operator "onnx.Unsqueeze"(%arg3) {torch.onnx.axes = [0 : si64]} : (!torch.vtensor<[],si64>) -> !torch.vtensor<[1],si64> - %6 = torch.operator "onnx.Unsqueeze"(%4) {torch.onnx.axes = [0 : si64]} : (!torch.vtensor<[],si64>) -> !torch.vtensor<[1],si64> - %7 = torch.operator "onnx.Slice"(%2, %5, %6) : (!torch.vtensor<[5],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[?],f32> - %8 = torch.operator "onnx.Add"(%arg5, %7) : (!torch.vtensor<[1],f32>, !torch.vtensor<[?],f32>) -> !torch.vtensor<[1],f32> - %9 = torch.operator "onnx.Identity"(%8) : (!torch.vtensor<[1],f32>) -> !torch.vtensor<[1],f32> - torch.operator_terminator %1, %8, %9 : !torch.vtensor<[],i1>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32> + %5 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> + %6 = torch.operator "onnx.Unsqueeze"(%arg3, %5) : (!torch.vtensor<[],si64>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[1],si64> + %7 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> + %8 = torch.operator "onnx.Unsqueeze"(%4, %7) : (!torch.vtensor<[],si64>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[1],si64> + %9 = torch.operator "onnx.Slice"(%2, %6, %8) : (!torch.vtensor<[5],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[?],f32> + %10 = torch.operator "onnx.Add"(%arg5, %9) : (!torch.vtensor<[1],f32>, !torch.vtensor<[?],f32>) -> !torch.vtensor<[1],f32> + %11 = torch.operator "onnx.Identity"(%10) : (!torch.vtensor<[1],f32>) -> !torch.vtensor<[1],f32> + torch.operator_terminator %1, %10, %11 : !torch.vtensor<[],i1>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32> } return %0#0, %0#1 : !torch.vtensor<[1],f32>, !torch.vtensor<[5,1],f32> } diff --git a/iree_tests/onnx/node/generated/test_lrn/model.mlir b/iree_tests/onnx/node/generated/test_lrn/model.mlir index 2ef117b1a..7eac9243f 100644 --- a/iree_tests/onnx/node/generated/test_lrn/model.mlir +++ b/iree_tests/onnx/node/generated/test_lrn/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_lrn(%arg0: !torch.vtensor<[5,5,5,5],f32>) -> !torch.vtensor<[5,5,5,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_lrn(%arg0: !torch.vtensor<[5,5,5,5],f32>) -> !torch.vtensor<[5,5,5,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LRN"(%arg0) {torch.onnx.alpha = 2.000000e-04 : f32, torch.onnx.beta = 5.000000e-01 : f32, torch.onnx.bias = 2.000000e+00 : f32, torch.onnx.size = 3 : si64} : (!torch.vtensor<[5,5,5,5],f32>) -> !torch.vtensor<[5,5,5,5],f32> return %0 : !torch.vtensor<[5,5,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_lrn_default/model.mlir b/iree_tests/onnx/node/generated/test_lrn_default/model.mlir index 9a9d0d7e7..09db0bfa5 100644 --- a/iree_tests/onnx/node/generated/test_lrn_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_lrn_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_lrn_default(%arg0: !torch.vtensor<[5,5,5,5],f32>) -> !torch.vtensor<[5,5,5,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_lrn_default(%arg0: !torch.vtensor<[5,5,5,5],f32>) -> !torch.vtensor<[5,5,5,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.LRN"(%arg0) {torch.onnx.size = 3 : si64} : (!torch.vtensor<[5,5,5,5],f32>) -> !torch.vtensor<[5,5,5,5],f32> return %0 : !torch.vtensor<[5,5,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_lstm_batchwise/model.mlir b/iree_tests/onnx/node/generated/test_lstm_batchwise/model.mlir index c8eced758..ed44b0b40 100644 --- a/iree_tests/onnx/node/generated/test_lstm_batchwise/model.mlir +++ b/iree_tests/onnx/node/generated/test_lstm_batchwise/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_lstm_batchwise(%arg0: !torch.vtensor<[3,1,2],f32>, %arg1: !torch.vtensor<[1,28,2],f32>, %arg2: !torch.vtensor<[1,28,7],f32>) -> (!torch.vtensor<[3,1,1,7],f32>, !torch.vtensor<[3,1,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_lstm_batchwise(%arg0: !torch.vtensor<[3,1,2],f32>, %arg1: !torch.vtensor<[1,28,2],f32>, %arg2: !torch.vtensor<[1,28,7],f32>) -> (!torch.vtensor<[3,1,1,7],f32>, !torch.vtensor<[3,1,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.LSTM"(%arg0, %arg1, %arg2) {torch.onnx.hidden_size = 7 : si64, torch.onnx.layout = 1 : si64} : (!torch.vtensor<[3,1,2],f32>, !torch.vtensor<[1,28,2],f32>, !torch.vtensor<[1,28,7],f32>) -> (!torch.vtensor<[3,1,1,7],f32>, !torch.vtensor<[3,1,7],f32>) return %0#0, %0#1 : !torch.vtensor<[3,1,1,7],f32>, !torch.vtensor<[3,1,7],f32> diff --git a/iree_tests/onnx/node/generated/test_lstm_defaults/model.mlir b/iree_tests/onnx/node/generated/test_lstm_defaults/model.mlir index 7754393c3..f91dc57ed 100644 --- a/iree_tests/onnx/node/generated/test_lstm_defaults/model.mlir +++ b/iree_tests/onnx/node/generated/test_lstm_defaults/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_lstm_defaults(%arg0: !torch.vtensor<[1,3,2],f32>, %arg1: !torch.vtensor<[1,12,2],f32>, %arg2: !torch.vtensor<[1,12,3],f32>) -> !torch.vtensor<[1,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_lstm_defaults(%arg0: !torch.vtensor<[1,3,2],f32>, %arg1: !torch.vtensor<[1,12,2],f32>, %arg2: !torch.vtensor<[1,12,3],f32>) -> !torch.vtensor<[1,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.LSTM"(%arg0, %arg1, %arg2) {torch.onnx.hidden_size = 3 : si64} : (!torch.vtensor<[1,3,2],f32>, !torch.vtensor<[1,12,2],f32>, !torch.vtensor<[1,12,3],f32>) -> (!torch.none, !torch.vtensor<[1,3,3],f32>) return %0#1 : !torch.vtensor<[1,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_lstm_with_initial_bias/model.mlir b/iree_tests/onnx/node/generated/test_lstm_with_initial_bias/model.mlir index 170e63154..fcfa9a1ab 100644 --- a/iree_tests/onnx/node/generated/test_lstm_with_initial_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_lstm_with_initial_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_lstm_with_initial_bias(%arg0: !torch.vtensor<[1,3,3],f32>, %arg1: !torch.vtensor<[1,16,3],f32>, %arg2: !torch.vtensor<[1,16,4],f32>, %arg3: !torch.vtensor<[1,32],f32>) -> !torch.vtensor<[1,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_lstm_with_initial_bias(%arg0: !torch.vtensor<[1,3,3],f32>, %arg1: !torch.vtensor<[1,16,3],f32>, %arg2: !torch.vtensor<[1,16,4],f32>, %arg3: !torch.vtensor<[1,32],f32>) -> !torch.vtensor<[1,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.LSTM"(%arg0, %arg1, %arg2, %arg3) {torch.onnx.hidden_size = 4 : si64} : (!torch.vtensor<[1,3,3],f32>, !torch.vtensor<[1,16,3],f32>, !torch.vtensor<[1,16,4],f32>, !torch.vtensor<[1,32],f32>) -> (!torch.none, !torch.vtensor<[1,3,4],f32>) return %0#1 : !torch.vtensor<[1,3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_lstm_with_peepholes/model.mlir b/iree_tests/onnx/node/generated/test_lstm_with_peepholes/model.mlir index 86447de13..38c1a0f92 100644 --- a/iree_tests/onnx/node/generated/test_lstm_with_peepholes/model.mlir +++ b/iree_tests/onnx/node/generated/test_lstm_with_peepholes/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_lstm_with_peepholes(%arg0: !torch.vtensor<[1,2,4],f32>, %arg1: !torch.vtensor<[1,12,4],f32>, %arg2: !torch.vtensor<[1,12,3],f32>, %arg3: !torch.vtensor<[1,24],f32>, %arg4: !torch.vtensor<[2],si32>, %arg5: !torch.vtensor<[1,2,3],f32>, %arg6: !torch.vtensor<[1,2,3],f32>, %arg7: !torch.vtensor<[1,9],f32>) -> !torch.vtensor<[1,2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_lstm_with_peepholes(%arg0: !torch.vtensor<[1,2,4],f32>, %arg1: !torch.vtensor<[1,12,4],f32>, %arg2: !torch.vtensor<[1,12,3],f32>, %arg3: !torch.vtensor<[1,24],f32>, %arg4: !torch.vtensor<[2],si32>, %arg5: !torch.vtensor<[1,2,3],f32>, %arg6: !torch.vtensor<[1,2,3],f32>, %arg7: !torch.vtensor<[1,9],f32>) -> !torch.vtensor<[1,2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.LSTM"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7) {torch.onnx.hidden_size = 3 : si64} : (!torch.vtensor<[1,2,4],f32>, !torch.vtensor<[1,12,4],f32>, !torch.vtensor<[1,12,3],f32>, !torch.vtensor<[1,24],f32>, !torch.vtensor<[2],si32>, !torch.vtensor<[1,2,3],f32>, !torch.vtensor<[1,2,3],f32>, !torch.vtensor<[1,9],f32>) -> (!torch.none, !torch.vtensor<[1,2,3],f32>) return %0#1 : !torch.vtensor<[1,2,3],f32> diff --git a/iree_tests/onnx/node/generated/test_matmul_2d/model.mlir b/iree_tests/onnx/node/generated/test_matmul_2d/model.mlir index 36a62ac8e..e228738f7 100644 --- a/iree_tests/onnx/node/generated/test_matmul_2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_matmul_2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_matmul_2d(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[4,3],f32>) -> !torch.vtensor<[3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_matmul_2d(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[4,3],f32>) -> !torch.vtensor<[3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MatMul"(%arg0, %arg1) : (!torch.vtensor<[3,4],f32>, !torch.vtensor<[4,3],f32>) -> !torch.vtensor<[3,3],f32> return %0 : !torch.vtensor<[3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_matmul_3d/model.mlir b/iree_tests/onnx/node/generated/test_matmul_3d/model.mlir index 7a34c0a90..24118e396 100644 --- a/iree_tests/onnx/node/generated/test_matmul_3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_matmul_3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_matmul_3d(%arg0: !torch.vtensor<[2,3,4],f32>, %arg1: !torch.vtensor<[2,4,3],f32>) -> !torch.vtensor<[2,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_matmul_3d(%arg0: !torch.vtensor<[2,3,4],f32>, %arg1: !torch.vtensor<[2,4,3],f32>) -> !torch.vtensor<[2,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MatMul"(%arg0, %arg1) : (!torch.vtensor<[2,3,4],f32>, !torch.vtensor<[2,4,3],f32>) -> !torch.vtensor<[2,3,3],f32> return %0 : !torch.vtensor<[2,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_matmul_4d/model.mlir b/iree_tests/onnx/node/generated/test_matmul_4d/model.mlir index 4bcca465d..332a6fa24 100644 --- a/iree_tests/onnx/node/generated/test_matmul_4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_matmul_4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_matmul_4d(%arg0: !torch.vtensor<[1,2,3,4],f32>, %arg1: !torch.vtensor<[1,2,4,3],f32>) -> !torch.vtensor<[1,2,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_matmul_4d(%arg0: !torch.vtensor<[1,2,3,4],f32>, %arg1: !torch.vtensor<[1,2,4,3],f32>) -> !torch.vtensor<[1,2,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MatMul"(%arg0, %arg1) : (!torch.vtensor<[1,2,3,4],f32>, !torch.vtensor<[1,2,4,3],f32>) -> !torch.vtensor<[1,2,3,3],f32> return %0 : !torch.vtensor<[1,2,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_matmulinteger/model.mlir b/iree_tests/onnx/node/generated/test_matmulinteger/model.mlir index 5e440b536..0185cab59 100644 --- a/iree_tests/onnx/node/generated/test_matmulinteger/model.mlir +++ b/iree_tests/onnx/node/generated/test_matmulinteger/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_matmulinteger(%arg0: !torch.vtensor<[4,3],ui8>, %arg1: !torch.vtensor<[3,2],ui8>, %arg2: !torch.vtensor<[1],ui8>, %arg3: !torch.vtensor<[1],ui8>) -> !torch.vtensor<[4,2],si32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_matmulinteger(%arg0: !torch.vtensor<[4,3],ui8>, %arg1: !torch.vtensor<[3,2],ui8>, %arg2: !torch.vtensor<[1],ui8>, %arg3: !torch.vtensor<[1],ui8>) -> !torch.vtensor<[4,2],si32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MatMulInteger"(%arg0, %arg1, %arg2, %arg3) : (!torch.vtensor<[4,3],ui8>, !torch.vtensor<[3,2],ui8>, !torch.vtensor<[1],ui8>, !torch.vtensor<[1],ui8>) -> !torch.vtensor<[4,2],si32> return %0 : !torch.vtensor<[4,2],si32> diff --git a/iree_tests/onnx/node/generated/test_max_example/model.mlir b/iree_tests/onnx/node/generated/test_max_example/model.mlir index a9408739b..da9b7e8ea 100644 --- a/iree_tests/onnx/node/generated/test_max_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_max_float16/model.mlir b/iree_tests/onnx/node/generated/test_max_float16/model.mlir index 129b3f080..d5dfc794b 100644 --- a/iree_tests/onnx/node/generated/test_max_float16/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_float16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_float16(%arg0: !torch.vtensor<[3],f16>, %arg1: !torch.vtensor<[3],f16>) -> !torch.vtensor<[3],f16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_float16(%arg0: !torch.vtensor<[3],f16>, %arg1: !torch.vtensor<[3],f16>) -> !torch.vtensor<[3],f16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],f16>, !torch.vtensor<[3],f16>) -> !torch.vtensor<[3],f16> return %0 : !torch.vtensor<[3],f16> diff --git a/iree_tests/onnx/node/generated/test_max_float32/model.mlir b/iree_tests/onnx/node/generated/test_max_float32/model.mlir index 9decf58e5..173cc0cc5 100644 --- a/iree_tests/onnx/node/generated/test_max_float32/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_float32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_float32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_float32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_max_float64/model.mlir b/iree_tests/onnx/node/generated/test_max_float64/model.mlir index 569f60743..5c61c6d57 100644 --- a/iree_tests/onnx/node/generated/test_max_float64/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_float64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_float64(%arg0: !torch.vtensor<[3],f64>, %arg1: !torch.vtensor<[3],f64>) -> !torch.vtensor<[3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_float64(%arg0: !torch.vtensor<[3],f64>, %arg1: !torch.vtensor<[3],f64>) -> !torch.vtensor<[3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],f64>, !torch.vtensor<[3],f64>) -> !torch.vtensor<[3],f64> return %0 : !torch.vtensor<[3],f64> diff --git a/iree_tests/onnx/node/generated/test_max_int16/model.mlir b/iree_tests/onnx/node/generated/test_max_int16/model.mlir index 4ee89c9a0..395d7ed5b 100644 --- a/iree_tests/onnx/node/generated/test_max_int16/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_int16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_int16(%arg0: !torch.vtensor<[3],si16>, %arg1: !torch.vtensor<[3],si16>) -> !torch.vtensor<[3],si16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_int16(%arg0: !torch.vtensor<[3],si16>, %arg1: !torch.vtensor<[3],si16>) -> !torch.vtensor<[3],si16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],si16>, !torch.vtensor<[3],si16>) -> !torch.vtensor<[3],si16> return %0 : !torch.vtensor<[3],si16> diff --git a/iree_tests/onnx/node/generated/test_max_int32/model.mlir b/iree_tests/onnx/node/generated/test_max_int32/model.mlir index 35df1ed2a..0d1d09b3c 100644 --- a/iree_tests/onnx/node/generated/test_max_int32/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_int32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_int32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_int32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],si32>, !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> return %0 : !torch.vtensor<[3],si32> diff --git a/iree_tests/onnx/node/generated/test_max_int64/model.mlir b/iree_tests/onnx/node/generated/test_max_int64/model.mlir index 6e9abe49c..4e3d753d2 100644 --- a/iree_tests/onnx/node/generated/test_max_int64/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_int64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_int64(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_int64(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> return %0 : !torch.vtensor<[3],si64> diff --git a/iree_tests/onnx/node/generated/test_max_int8/model.mlir b/iree_tests/onnx/node/generated/test_max_int8/model.mlir index 75ac35358..e33c344c0 100644 --- a/iree_tests/onnx/node/generated/test_max_int8/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_int8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_int8(%arg0: !torch.vtensor<[3],si8>, %arg1: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_int8(%arg0: !torch.vtensor<[3],si8>, %arg1: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],si8>, !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> return %0 : !torch.vtensor<[3],si8> diff --git a/iree_tests/onnx/node/generated/test_max_one_input/model.mlir b/iree_tests/onnx/node/generated/test_max_one_input/model.mlir index 5933c708d..864549cd1 100644 --- a/iree_tests/onnx/node/generated/test_max_one_input/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_one_input/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_max_two_inputs/model.mlir b/iree_tests/onnx/node/generated/test_max_two_inputs/model.mlir index 454e43eda..2c8c2abda 100644 --- a/iree_tests/onnx/node/generated/test_max_two_inputs/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_two_inputs/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_max_uint16/model.mlir b/iree_tests/onnx/node/generated/test_max_uint16/model.mlir index 946ca999c..c0e570706 100644 --- a/iree_tests/onnx/node/generated/test_max_uint16/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_uint16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],ui16>, !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> return %0 : !torch.vtensor<[3],ui16> diff --git a/iree_tests/onnx/node/generated/test_max_uint32/model.mlir b/iree_tests/onnx/node/generated/test_max_uint32/model.mlir index 9cf24d148..de6cd8532 100644 --- a/iree_tests/onnx/node/generated/test_max_uint32/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_uint32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],ui32>, !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> return %0 : !torch.vtensor<[3],ui32> diff --git a/iree_tests/onnx/node/generated/test_max_uint64/model.mlir b/iree_tests/onnx/node/generated/test_max_uint64/model.mlir index 1455c2712..3e0943e5f 100644 --- a/iree_tests/onnx/node/generated/test_max_uint64/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_uint64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],ui64>, !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> return %0 : !torch.vtensor<[3],ui64> diff --git a/iree_tests/onnx/node/generated/test_max_uint8/model.mlir b/iree_tests/onnx/node/generated/test_max_uint8/model.mlir index 21413b040..1cede0108 100644 --- a/iree_tests/onnx/node/generated/test_max_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_max_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_max_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_max_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Max"(%arg0, %arg1) : (!torch.vtensor<[3],ui8>, !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> return %0 : !torch.vtensor<[3],ui8> diff --git a/iree_tests/onnx/node/generated/test_maxpool_1d_default/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_1d_default/model.mlir index 54789c5b6..3b4475533 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_1d_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_1d_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_1d_default(%arg0: !torch.vtensor<[1,3,32],f32>) -> !torch.vtensor<[1,3,31],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_1d_default(%arg0: !torch.vtensor<[1,3,32],f32>) -> !torch.vtensor<[1,3,31],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [2 : si64]} : (!torch.vtensor<[1,3,32],f32>) -> !torch.vtensor<[1,3,31],f32> return %0 : !torch.vtensor<[1,3,31],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_ceil/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_ceil/model.mlir index eb4e9bf38..b3c861687 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_ceil/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_ceil/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_ceil(%arg0: !torch.vtensor<[1,1,4,4],f32>) -> !torch.vtensor<[1,1,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_ceil(%arg0: !torch.vtensor<[1,1,4,4],f32>) -> !torch.vtensor<[1,1,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.ceil_mode = 1 : si64, torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,4,4],f32>) -> !torch.vtensor<[1,1,2,2],f32> return %0 : !torch.vtensor<[1,1,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_default/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_default/model.mlir index 73cec36f9..221855957 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_default(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,31,31],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_default(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,31,31],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,31,31],f32> return %0 : !torch.vtensor<[1,3,31,31],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_dilations/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_dilations/model.mlir index 720bbdb11..1eb4abf4d 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_dilations/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_dilations/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_dilations(%arg0: !torch.vtensor<[1,1,4,4],f32>) -> !torch.vtensor<[1,1,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_dilations(%arg0: !torch.vtensor<[1,1,4,4],f32>) -> !torch.vtensor<[1,1,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.dilations = [2 : si64, 2 : si64], torch.onnx.kernel_shape = [2 : si64, 2 : si64], torch.onnx.strides = [1 : si64, 1 : si64]} : (!torch.vtensor<[1,1,4,4],f32>) -> !torch.vtensor<[1,1,2,2],f32> return %0 : !torch.vtensor<[1,1,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_pads/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_pads/model.mlir index 127595c84..2069e155d 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_pads/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_pads/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_pads(%arg0: !torch.vtensor<[1,3,28,28],f32>) -> !torch.vtensor<[1,3,30,30],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_pads(%arg0: !torch.vtensor<[1,3,28,28],f32>) -> !torch.vtensor<[1,3,30,30],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.pads = [2 : si64, 2 : si64, 2 : si64, 2 : si64]} : (!torch.vtensor<[1,3,28,28],f32>) -> !torch.vtensor<[1,3,30,30],f32> return %0 : !torch.vtensor<[1,3,30,30],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_pads/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_pads/model.mlir index d2ec889e7..fbe9bcc76 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_pads/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_pads/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_precomputed_pads(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_precomputed_pads(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [5 : si64, 5 : si64], torch.onnx.pads = [2 : si64, 2 : si64, 2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,5,5],f32> return %0 : !torch.vtensor<[1,1,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_same_upper/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_same_upper/model.mlir index a808522be..d713a24ed 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_same_upper/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_same_upper/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_precomputed_same_upper(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_precomputed_same_upper(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.auto_pad = "SAME_UPPER", torch.onnx.kernel_shape = [3 : si64, 3 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,3,3],f32> return %0 : !torch.vtensor<[1,1,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_strides/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_strides/model.mlir index 63fbefd53..0f484da05 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_strides/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_precomputed_strides/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_precomputed_strides(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_precomputed_strides(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [2 : si64, 2 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,5,5],f32>) -> !torch.vtensor<[1,1,2,2],f32> return %0 : !torch.vtensor<[1,1,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_same_lower/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_same_lower/model.mlir index a523570aa..9c0e83857 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_same_lower/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_same_lower/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_same_lower(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_same_lower(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.auto_pad = "SAME_LOWER", torch.onnx.kernel_shape = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> return %0 : !torch.vtensor<[1,3,32,32],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_same_upper/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_same_upper/model.mlir index f4ad24f6e..b632dc773 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_same_upper/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_same_upper/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_same_upper(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_same_upper(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.auto_pad = "SAME_UPPER", torch.onnx.kernel_shape = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,32,32],f32> return %0 : !torch.vtensor<[1,3,32,32],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_strides/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_strides/model.mlir index 0cf937499..05b442003 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_strides/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_strides/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_strides(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,10,10],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_strides(%arg0: !torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,10,10],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [5 : si64, 5 : si64], torch.onnx.strides = [3 : si64, 3 : si64]} : (!torch.vtensor<[1,3,32,32],f32>) -> !torch.vtensor<[1,3,10,10],f32> return %0 : !torch.vtensor<[1,3,10,10],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_2d_uint8/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_2d_uint8/model.mlir index 3a2ff000c..d59c8335c 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_2d_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_2d_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_2d_uint8(%arg0: !torch.vtensor<[1,1,5,5],ui8>) -> !torch.vtensor<[1,1,5,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_2d_uint8(%arg0: !torch.vtensor<[1,1,5,5],ui8>) -> !torch.vtensor<[1,1,5,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [5 : si64, 5 : si64], torch.onnx.pads = [2 : si64, 2 : si64, 2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,5,5],ui8>) -> !torch.vtensor<[1,1,5,5],ui8> return %0 : !torch.vtensor<[1,1,5,5],ui8> diff --git a/iree_tests/onnx/node/generated/test_maxpool_3d_default/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_3d_default/model.mlir index cb6fe9b99..e3eb630dd 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_3d_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_3d_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_3d_default(%arg0: !torch.vtensor<[1,3,32,32,32],f32>) -> !torch.vtensor<[1,3,31,31,31],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_3d_default(%arg0: !torch.vtensor<[1,3,32,32,32],f32>) -> !torch.vtensor<[1,3,31,31,31],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [2 : si64, 2 : si64, 2 : si64]} : (!torch.vtensor<[1,3,32,32,32],f32>) -> !torch.vtensor<[1,3,31,31,31],f32> return %0 : !torch.vtensor<[1,3,31,31,31],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_3d_dilations/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_3d_dilations/model.mlir index 5b24a5e8d..56b1b51ed 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_3d_dilations/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_3d_dilations/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_3d_dilations(%arg0: !torch.vtensor<[1,1,4,4,4],f32>) -> !torch.vtensor<[1,1,2,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_3d_dilations(%arg0: !torch.vtensor<[1,1,4,4,4],f32>) -> !torch.vtensor<[1,1,2,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.dilations = [2 : si64, 2 : si64, 2 : si64], torch.onnx.kernel_shape = [2 : si64, 2 : si64, 2 : si64], torch.onnx.strides = [1 : si64, 1 : si64, 1 : si64]} : (!torch.vtensor<[1,1,4,4,4],f32>) -> !torch.vtensor<[1,1,2,2,2],f32> return %0 : !torch.vtensor<[1,1,2,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl/model.mlir index 582dc1689..f7e4d0de0 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_3d_dilations_use_ref_impl(%arg0: !torch.vtensor<[1,1,4,4,4],f32>) -> !torch.vtensor<[1,1,2,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_3d_dilations_use_ref_impl(%arg0: !torch.vtensor<[1,1,4,4,4],f32>) -> !torch.vtensor<[1,1,2,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.dilations = [2 : si64, 2 : si64, 2 : si64], torch.onnx.kernel_shape = [2 : si64, 2 : si64, 2 : si64], torch.onnx.strides = [1 : si64, 1 : si64, 1 : si64]} : (!torch.vtensor<[1,1,4,4,4],f32>) -> !torch.vtensor<[1,1,2,2,2],f32> return %0 : !torch.vtensor<[1,1,2,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl_large/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl_large/model.mlir index cb23de8bf..7972c25d7 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl_large/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_3d_dilations_use_ref_impl_large/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_3d_dilations_use_ref_impl_large(%arg0: !torch.vtensor<[1,1,32,32,32],f32>) -> !torch.vtensor<[1,1,9,9,9],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_3d_dilations_use_ref_impl_large(%arg0: !torch.vtensor<[1,1,32,32,32],f32>) -> !torch.vtensor<[1,1,9,9,9],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.ceil_mode = 1 : si64, torch.onnx.dilations = [2 : si64, 2 : si64, 2 : si64], torch.onnx.kernel_shape = [5 : si64, 5 : si64, 5 : si64], torch.onnx.strides = [3 : si64, 3 : si64, 3 : si64]} : (!torch.vtensor<[1,1,32,32,32],f32>) -> !torch.vtensor<[1,1,9,9,9],f32> return %0 : !torch.vtensor<[1,1,9,9,9],f32> diff --git a/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_pads/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_pads/model.mlir index 035630b8b..1935284d7 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_pads/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_pads/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_with_argmax_2d_precomputed_pads(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> (!torch.vtensor<[1,1,5,5],f32>, !torch.vtensor<[1,1,5,5],si64>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_with_argmax_2d_precomputed_pads(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> (!torch.vtensor<[1,1,5,5],f32>, !torch.vtensor<[1,1,5,5],si64>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [5 : si64, 5 : si64], torch.onnx.pads = [2 : si64, 2 : si64, 2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,5,5],f32>) -> (!torch.vtensor<[1,1,5,5],f32>, !torch.vtensor<[1,1,5,5],si64>) return %0#0, %0#1 : !torch.vtensor<[1,1,5,5],f32>, !torch.vtensor<[1,1,5,5],si64> diff --git a/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides/model.mlir b/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides/model.mlir index cd4e2095a..f1577d90a 100644 --- a/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxpool_with_argmax_2d_precomputed_strides(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> (!torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[1,1,2,2],si64>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 12 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxpool_with_argmax_2d_precomputed_strides(%arg0: !torch.vtensor<[1,1,5,5],f32>) -> (!torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[1,1,2,2],si64>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.MaxPool"(%arg0) {torch.onnx.kernel_shape = [2 : si64, 2 : si64], torch.onnx.storage_order = 1 : si64, torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,5,5],f32>) -> (!torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[1,1,2,2],si64>) return %0#0, %0#1 : !torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[1,1,2,2],si64> diff --git a/iree_tests/onnx/node/generated/test_maxunpool_export_with_output_shape/model.mlir b/iree_tests/onnx/node/generated/test_maxunpool_export_with_output_shape/model.mlir index fc4c34a6a..9d0b221e8 100644 --- a/iree_tests/onnx/node/generated/test_maxunpool_export_with_output_shape/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxunpool_export_with_output_shape/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxunpool_export_with_output_shape(%arg0: !torch.vtensor<[1,1,2,2],f32>, %arg1: !torch.vtensor<[1,1,2,2],si64>, %arg2: !torch.vtensor<[4],si64>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxunpool_export_with_output_shape(%arg0: !torch.vtensor<[1,1,2,2],f32>, %arg1: !torch.vtensor<[1,1,2,2],si64>, %arg2: !torch.vtensor<[4],si64>) -> !torch.vtensor<[1,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxUnpool"(%arg0, %arg1, %arg2) {torch.onnx.kernel_shape = [2 : si64, 2 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[1,1,2,2],si64>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[1,1,5,5],f32> return %0 : !torch.vtensor<[1,1,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_maxunpool_export_without_output_shape/model.mlir b/iree_tests/onnx/node/generated/test_maxunpool_export_without_output_shape/model.mlir index 71797b15f..7ef3c96ae 100644 --- a/iree_tests/onnx/node/generated/test_maxunpool_export_without_output_shape/model.mlir +++ b/iree_tests/onnx/node/generated/test_maxunpool_export_without_output_shape/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_maxunpool_export_without_output_shape(%arg0: !torch.vtensor<[1,1,2,2],f32>, %arg1: !torch.vtensor<[1,1,2,2],si64>) -> !torch.vtensor<[1,1,4,4],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_maxunpool_export_without_output_shape(%arg0: !torch.vtensor<[1,1,2,2],f32>, %arg1: !torch.vtensor<[1,1,2,2],si64>) -> !torch.vtensor<[1,1,4,4],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MaxUnpool"(%arg0, %arg1) {torch.onnx.kernel_shape = [2 : si64, 2 : si64], torch.onnx.strides = [2 : si64, 2 : si64]} : (!torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[1,1,2,2],si64>) -> !torch.vtensor<[1,1,4,4],f32> return %0 : !torch.vtensor<[1,1,4,4],f32> diff --git a/iree_tests/onnx/node/generated/test_mean_example/model.mlir b/iree_tests/onnx/node/generated/test_mean_example/model.mlir index dcb7ab6a5..f17d8caea 100644 --- a/iree_tests/onnx/node/generated/test_mean_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_mean_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mean_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mean_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mean"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_mean_one_input/model.mlir b/iree_tests/onnx/node/generated/test_mean_one_input/model.mlir index 8407974cf..488b92bd4 100644 --- a/iree_tests/onnx/node/generated/test_mean_one_input/model.mlir +++ b/iree_tests/onnx/node/generated/test_mean_one_input/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mean_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mean_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mean"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_mean_two_inputs/model.mlir b/iree_tests/onnx/node/generated/test_mean_two_inputs/model.mlir index 05f492b03..ac3a33d34 100644 --- a/iree_tests/onnx/node/generated/test_mean_two_inputs/model.mlir +++ b/iree_tests/onnx/node/generated/test_mean_two_inputs/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mean_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mean_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mean"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_min_example/model.mlir b/iree_tests/onnx/node/generated/test_min_example/model.mlir index 605b7a61c..aa6728413 100644 --- a/iree_tests/onnx/node/generated/test_min_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_min_float16/model.mlir b/iree_tests/onnx/node/generated/test_min_float16/model.mlir index b2ccc0d20..48e30f42a 100644 --- a/iree_tests/onnx/node/generated/test_min_float16/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_float16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_float16(%arg0: !torch.vtensor<[3],f16>, %arg1: !torch.vtensor<[3],f16>) -> !torch.vtensor<[3],f16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_float16(%arg0: !torch.vtensor<[3],f16>, %arg1: !torch.vtensor<[3],f16>) -> !torch.vtensor<[3],f16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],f16>, !torch.vtensor<[3],f16>) -> !torch.vtensor<[3],f16> return %0 : !torch.vtensor<[3],f16> diff --git a/iree_tests/onnx/node/generated/test_min_float32/model.mlir b/iree_tests/onnx/node/generated/test_min_float32/model.mlir index 6979aa36f..64fe439ae 100644 --- a/iree_tests/onnx/node/generated/test_min_float32/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_float32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_float32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_float32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_min_float64/model.mlir b/iree_tests/onnx/node/generated/test_min_float64/model.mlir index f5a756709..df867f3df 100644 --- a/iree_tests/onnx/node/generated/test_min_float64/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_float64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_float64(%arg0: !torch.vtensor<[3],f64>, %arg1: !torch.vtensor<[3],f64>) -> !torch.vtensor<[3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_float64(%arg0: !torch.vtensor<[3],f64>, %arg1: !torch.vtensor<[3],f64>) -> !torch.vtensor<[3],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],f64>, !torch.vtensor<[3],f64>) -> !torch.vtensor<[3],f64> return %0 : !torch.vtensor<[3],f64> diff --git a/iree_tests/onnx/node/generated/test_min_int16/model.mlir b/iree_tests/onnx/node/generated/test_min_int16/model.mlir index 061664cb9..92cfa5c4f 100644 --- a/iree_tests/onnx/node/generated/test_min_int16/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_int16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_int16(%arg0: !torch.vtensor<[3],si16>, %arg1: !torch.vtensor<[3],si16>) -> !torch.vtensor<[3],si16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_int16(%arg0: !torch.vtensor<[3],si16>, %arg1: !torch.vtensor<[3],si16>) -> !torch.vtensor<[3],si16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],si16>, !torch.vtensor<[3],si16>) -> !torch.vtensor<[3],si16> return %0 : !torch.vtensor<[3],si16> diff --git a/iree_tests/onnx/node/generated/test_min_int32/model.mlir b/iree_tests/onnx/node/generated/test_min_int32/model.mlir index 25517b1a1..1a858352c 100644 --- a/iree_tests/onnx/node/generated/test_min_int32/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_int32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_int32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_int32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],si32>, !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> return %0 : !torch.vtensor<[3],si32> diff --git a/iree_tests/onnx/node/generated/test_min_int64/model.mlir b/iree_tests/onnx/node/generated/test_min_int64/model.mlir index e9c8f918c..f802e0e07 100644 --- a/iree_tests/onnx/node/generated/test_min_int64/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_int64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_int64(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_int64(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> return %0 : !torch.vtensor<[3],si64> diff --git a/iree_tests/onnx/node/generated/test_min_int8/model.mlir b/iree_tests/onnx/node/generated/test_min_int8/model.mlir index f65451f2b..9fbf2dae9 100644 --- a/iree_tests/onnx/node/generated/test_min_int8/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_int8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_int8(%arg0: !torch.vtensor<[3],si8>, %arg1: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_int8(%arg0: !torch.vtensor<[3],si8>, %arg1: !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],si8>, !torch.vtensor<[3],si8>) -> !torch.vtensor<[3],si8> return %0 : !torch.vtensor<[3],si8> diff --git a/iree_tests/onnx/node/generated/test_min_one_input/model.mlir b/iree_tests/onnx/node/generated/test_min_one_input/model.mlir index 67ce1c5f0..057d9b70d 100644 --- a/iree_tests/onnx/node/generated/test_min_one_input/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_one_input/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_min_two_inputs/model.mlir b/iree_tests/onnx/node/generated/test_min_two_inputs/model.mlir index 2156ae83a..28e22f48a 100644 --- a/iree_tests/onnx/node/generated/test_min_two_inputs/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_two_inputs/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_min_uint16/model.mlir b/iree_tests/onnx/node/generated/test_min_uint16/model.mlir index f4360dfc1..24a5c7b7a 100644 --- a/iree_tests/onnx/node/generated/test_min_uint16/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_uint16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],ui16>, !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> return %0 : !torch.vtensor<[3],ui16> diff --git a/iree_tests/onnx/node/generated/test_min_uint32/model.mlir b/iree_tests/onnx/node/generated/test_min_uint32/model.mlir index f930a5a6a..118f73ca9 100644 --- a/iree_tests/onnx/node/generated/test_min_uint32/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_uint32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],ui32>, !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> return %0 : !torch.vtensor<[3],ui32> diff --git a/iree_tests/onnx/node/generated/test_min_uint64/model.mlir b/iree_tests/onnx/node/generated/test_min_uint64/model.mlir index ee2ef5636..00aa1de53 100644 --- a/iree_tests/onnx/node/generated/test_min_uint64/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_uint64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],ui64>, !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> return %0 : !torch.vtensor<[3],ui64> diff --git a/iree_tests/onnx/node/generated/test_min_uint8/model.mlir b/iree_tests/onnx/node/generated/test_min_uint8/model.mlir index dadf8fd76..0c4b1b243 100644 --- a/iree_tests/onnx/node/generated/test_min_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_min_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_min_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_min_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Min"(%arg0, %arg1) : (!torch.vtensor<[3],ui8>, !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> return %0 : !torch.vtensor<[3],ui8> diff --git a/iree_tests/onnx/node/generated/test_mod_broadcast/model.mlir b/iree_tests/onnx/node/generated/test_mod_broadcast/model.mlir index 628c1f151..07f253ba0 100644 --- a/iree_tests/onnx/node/generated/test_mod_broadcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_broadcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_broadcast(%arg0: !torch.vtensor<[3,2,5],si32>, %arg1: !torch.vtensor<[1],si32>) -> !torch.vtensor<[3,2,5],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_broadcast(%arg0: !torch.vtensor<[3,2,5],si32>, %arg1: !torch.vtensor<[1],si32>) -> !torch.vtensor<[3,2,5],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[3,2,5],si32>, !torch.vtensor<[1],si32>) -> !torch.vtensor<[3,2,5],si32> return %0 : !torch.vtensor<[3,2,5],si32> diff --git a/iree_tests/onnx/node/generated/test_mod_int64_fmod/model.mlir b/iree_tests/onnx/node/generated/test_mod_int64_fmod/model.mlir index 37d0fc51c..aec9c5c9b 100644 --- a/iree_tests/onnx/node/generated/test_mod_int64_fmod/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_int64_fmod/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_int64_fmod(%arg0: !torch.vtensor<[6],si64>, %arg1: !torch.vtensor<[6],si64>) -> !torch.vtensor<[6],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_int64_fmod(%arg0: !torch.vtensor<[6],si64>, %arg1: !torch.vtensor<[6],si64>) -> !torch.vtensor<[6],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) {torch.onnx.fmod = 1 : si64} : (!torch.vtensor<[6],si64>, !torch.vtensor<[6],si64>) -> !torch.vtensor<[6],si64> return %0 : !torch.vtensor<[6],si64> diff --git a/iree_tests/onnx/node/generated/test_mod_mixed_sign_float16/model.mlir b/iree_tests/onnx/node/generated/test_mod_mixed_sign_float16/model.mlir index 842fc8124..1fe6d11b2 100644 --- a/iree_tests/onnx/node/generated/test_mod_mixed_sign_float16/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_mixed_sign_float16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_mixed_sign_float16(%arg0: !torch.vtensor<[6],f16>, %arg1: !torch.vtensor<[6],f16>) -> !torch.vtensor<[6],f16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_mixed_sign_float16(%arg0: !torch.vtensor<[6],f16>, %arg1: !torch.vtensor<[6],f16>) -> !torch.vtensor<[6],f16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) {torch.onnx.fmod = 1 : si64} : (!torch.vtensor<[6],f16>, !torch.vtensor<[6],f16>) -> !torch.vtensor<[6],f16> return %0 : !torch.vtensor<[6],f16> diff --git a/iree_tests/onnx/node/generated/test_mod_mixed_sign_float32/model.mlir b/iree_tests/onnx/node/generated/test_mod_mixed_sign_float32/model.mlir index 1f0598e2b..7ece27188 100644 --- a/iree_tests/onnx/node/generated/test_mod_mixed_sign_float32/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_mixed_sign_float32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_mixed_sign_float32(%arg0: !torch.vtensor<[6],f32>, %arg1: !torch.vtensor<[6],f32>) -> !torch.vtensor<[6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_mixed_sign_float32(%arg0: !torch.vtensor<[6],f32>, %arg1: !torch.vtensor<[6],f32>) -> !torch.vtensor<[6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) {torch.onnx.fmod = 1 : si64} : (!torch.vtensor<[6],f32>, !torch.vtensor<[6],f32>) -> !torch.vtensor<[6],f32> return %0 : !torch.vtensor<[6],f32> diff --git a/iree_tests/onnx/node/generated/test_mod_mixed_sign_float64/model.mlir b/iree_tests/onnx/node/generated/test_mod_mixed_sign_float64/model.mlir index b8187f722..8333bab09 100644 --- a/iree_tests/onnx/node/generated/test_mod_mixed_sign_float64/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_mixed_sign_float64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_mixed_sign_float64(%arg0: !torch.vtensor<[6],f64>, %arg1: !torch.vtensor<[6],f64>) -> !torch.vtensor<[6],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_mixed_sign_float64(%arg0: !torch.vtensor<[6],f64>, %arg1: !torch.vtensor<[6],f64>) -> !torch.vtensor<[6],f64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) {torch.onnx.fmod = 1 : si64} : (!torch.vtensor<[6],f64>, !torch.vtensor<[6],f64>) -> !torch.vtensor<[6],f64> return %0 : !torch.vtensor<[6],f64> diff --git a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int16/model.mlir b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int16/model.mlir index c4ceac1f9..42488f778 100644 --- a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int16/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_mixed_sign_int16(%arg0: !torch.vtensor<[6],si16>, %arg1: !torch.vtensor<[6],si16>) -> !torch.vtensor<[6],si16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_mixed_sign_int16(%arg0: !torch.vtensor<[6],si16>, %arg1: !torch.vtensor<[6],si16>) -> !torch.vtensor<[6],si16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[6],si16>, !torch.vtensor<[6],si16>) -> !torch.vtensor<[6],si16> return %0 : !torch.vtensor<[6],si16> diff --git a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int32/model.mlir b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int32/model.mlir index a1e27c7ae..b2cff3f43 100644 --- a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int32/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_mixed_sign_int32(%arg0: !torch.vtensor<[6],si32>, %arg1: !torch.vtensor<[6],si32>) -> !torch.vtensor<[6],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_mixed_sign_int32(%arg0: !torch.vtensor<[6],si32>, %arg1: !torch.vtensor<[6],si32>) -> !torch.vtensor<[6],si32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[6],si32>, !torch.vtensor<[6],si32>) -> !torch.vtensor<[6],si32> return %0 : !torch.vtensor<[6],si32> diff --git a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int64/model.mlir b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int64/model.mlir index 3037083f6..612eb2693 100644 --- a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int64/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_mixed_sign_int64(%arg0: !torch.vtensor<[6],si64>, %arg1: !torch.vtensor<[6],si64>) -> !torch.vtensor<[6],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_mixed_sign_int64(%arg0: !torch.vtensor<[6],si64>, %arg1: !torch.vtensor<[6],si64>) -> !torch.vtensor<[6],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[6],si64>, !torch.vtensor<[6],si64>) -> !torch.vtensor<[6],si64> return %0 : !torch.vtensor<[6],si64> diff --git a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int8/model.mlir b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int8/model.mlir index 6e15f4b62..25b393179 100644 --- a/iree_tests/onnx/node/generated/test_mod_mixed_sign_int8/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_mixed_sign_int8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_mixed_sign_int8(%arg0: !torch.vtensor<[6],si8>, %arg1: !torch.vtensor<[6],si8>) -> !torch.vtensor<[6],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_mixed_sign_int8(%arg0: !torch.vtensor<[6],si8>, %arg1: !torch.vtensor<[6],si8>) -> !torch.vtensor<[6],si8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[6],si8>, !torch.vtensor<[6],si8>) -> !torch.vtensor<[6],si8> return %0 : !torch.vtensor<[6],si8> diff --git a/iree_tests/onnx/node/generated/test_mod_uint16/model.mlir b/iree_tests/onnx/node/generated/test_mod_uint16/model.mlir index 6323b0e7f..47fc1cc27 100644 --- a/iree_tests/onnx/node/generated/test_mod_uint16/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_uint16/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_uint16(%arg0: !torch.vtensor<[3],ui16>, %arg1: !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[3],ui16>, !torch.vtensor<[3],ui16>) -> !torch.vtensor<[3],ui16> return %0 : !torch.vtensor<[3],ui16> diff --git a/iree_tests/onnx/node/generated/test_mod_uint32/model.mlir b/iree_tests/onnx/node/generated/test_mod_uint32/model.mlir index 7d973e1a2..ed16b9ba0 100644 --- a/iree_tests/onnx/node/generated/test_mod_uint32/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_uint32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_uint32(%arg0: !torch.vtensor<[3],ui32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[3],ui32>, !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],ui32> return %0 : !torch.vtensor<[3],ui32> diff --git a/iree_tests/onnx/node/generated/test_mod_uint64/model.mlir b/iree_tests/onnx/node/generated/test_mod_uint64/model.mlir index f06032ceb..477a0d656 100644 --- a/iree_tests/onnx/node/generated/test_mod_uint64/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_uint64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_uint64(%arg0: !torch.vtensor<[3],ui64>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[3],ui64>, !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],ui64> return %0 : !torch.vtensor<[3],ui64> diff --git a/iree_tests/onnx/node/generated/test_mod_uint8/model.mlir b/iree_tests/onnx/node/generated/test_mod_uint8/model.mlir index 7f613b782..b85ad4815 100644 --- a/iree_tests/onnx/node/generated/test_mod_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_mod_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mod_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mod_uint8(%arg0: !torch.vtensor<[3],ui8>, %arg1: !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mod"(%arg0, %arg1) : (!torch.vtensor<[3],ui8>, !torch.vtensor<[3],ui8>) -> !torch.vtensor<[3],ui8> return %0 : !torch.vtensor<[3],ui8> diff --git a/iree_tests/onnx/node/generated/test_mul/model.mlir b/iree_tests/onnx/node/generated/test_mul/model.mlir index c6491babb..7d0290ba5 100644 --- a/iree_tests/onnx/node/generated/test_mul/model.mlir +++ b/iree_tests/onnx/node/generated/test_mul/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mul(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mul(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mul"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_mul_bcast/model.mlir b/iree_tests/onnx/node/generated/test_mul_bcast/model.mlir index f5e8323df..8a3e2ef4d 100644 --- a/iree_tests/onnx/node/generated/test_mul_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_mul_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mul_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mul_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mul"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_mul_example/model.mlir b/iree_tests/onnx/node/generated/test_mul_example/model.mlir index 6276df7b8..8558817fc 100644 --- a/iree_tests/onnx/node/generated/test_mul_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_mul_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mul_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mul_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mul"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_mul_uint8/model.mlir b/iree_tests/onnx/node/generated/test_mul_uint8/model.mlir index 0964bc089..f1f3b3ea3 100644 --- a/iree_tests/onnx/node/generated/test_mul_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_mul_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mul_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mul_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Mul"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],ui8>, !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> return %0 : !torch.vtensor<[3,4,5],ui8> diff --git a/iree_tests/onnx/node/generated/test_mvn/model.mlir b/iree_tests/onnx/node/generated/test_mvn/model.mlir index 6e8f4a431..70cead9b7 100644 --- a/iree_tests/onnx/node/generated/test_mvn/model.mlir +++ b/iree_tests/onnx/node/generated/test_mvn/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mvn(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mvn(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.MeanVarianceNormalization"(%arg0) : (!torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> return %0 : !torch.vtensor<[3,3,3,1],f32> diff --git a/iree_tests/onnx/node/generated/test_mvn_expanded/model.mlir b/iree_tests/onnx/node/generated/test_mvn_expanded/model.mlir index 93098bdb8..e9af68b43 100644 --- a/iree_tests/onnx/node/generated/test_mvn_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_mvn_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_mvn_expanded(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_mvn_expanded(%arg0: !torch.vtensor<[3,3,3,1],f32>) -> !torch.vtensor<[3,3,3,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2.000000e+00> : tensor} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<9.99999971E-10> : tensor} : () -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_neg/model.mlir b/iree_tests/onnx/node/generated/test_neg/model.mlir index ed8f9a80e..bd0dd95a8 100644 --- a/iree_tests/onnx/node/generated/test_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_neg(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_neg(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Neg"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_neg_example/model.mlir b/iree_tests/onnx/node/generated/test_neg_example/model.mlir index 172de21fe..c07a8728a 100644 --- a/iree_tests/onnx/node/generated/test_neg_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_neg_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_neg_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_neg_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Neg"(%arg0) : (!torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> return %0 : !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NC/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NC/model.mlir index 33d95b78b..c92b3b363 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NC/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NC/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NC(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NC(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NC_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NC_expanded/model.mlir index 25fd6c05d..b099af936 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NC_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NC_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NC_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NC_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1/model.mlir index 0c6b82e31..eb62c5d83 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_expanded/model.mlir index 3e2e8db87..bbd03c747 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii/model.mlir index 3635acb1f..380c2cfd6 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_ii(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_ii(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii_expanded/model.mlir index 076f89656..6d9eebbee 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_ii_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_ii_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii/model.mlir index b25c07f64..744e0458e 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_mean_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_mean_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = -1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii_expanded/model.mlir index 0e1fb4059..56ca7a3a4 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_mean_weight_negative_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_mean_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_mean_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight/model.mlir index aa47a0f5a..013ba6877 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_weight(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_weight(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_expanded/model.mlir index df1d3c453..ab6b5e67b 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_weight_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_weight_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii/model.mlir index 0b3edfe4f..423871f81 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_weight_ii(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_weight_ii(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii_expanded/model.mlir index 8edf07289..3e49638ea 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1_weight_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1_weight_ii_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1_weight_ii_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2/model.mlir index 4afeca303..f11599a9f 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[3,6,6],f32> return %0 : !torch.vtensor<[3,6,6],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_expanded/model.mlir index 66360e1a1..a4cc169ef 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii/model.mlir index 95049ab24..c67291f46 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_no_weight_reduction_mean_ii(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_no_weight_reduction_mean_ii(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii_expanded/model.mlir index 5a217193d..2215660f8 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_no_weight_reduction_mean_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_no_weight_reduction_mean_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_no_weight_reduction_mean_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean/model.mlir index 0a87cc854..7010a94cb 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_reduction_mean(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_reduction_mean(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean_expanded/model.mlir index 4e270d1d8..f97c98a7d 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_mean_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_reduction_mean_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_reduction_mean_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum/model.mlir index 2093688f7..9616e1fe6 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_reduction_sum(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_reduction_sum(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum_expanded/model.mlir index 0a5aa1095..e3c1c0e25 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_reduction_sum_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_reduction_sum_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_reduction_sum_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight/model.mlir index cce0d63d2..e5bb7d7d1 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,6,6],f32> return %0 : !torch.vtensor<[3,6,6],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_expanded/model.mlir index e9cea12cb..fe04be92a 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,6,6],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean/model.mlir index b2af3d4bd..ccd099860 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight_reduction_mean(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight_reduction_mean(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean_expanded/model.mlir index cfb864a39..570f0cff1 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_mean_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight_reduction_mean_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight_reduction_mean_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum/model.mlir index 2a000fb77..977c4259b 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight_reduction_sum(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight_reduction_sum(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_expanded/model.mlir index 4def16729..114c63784 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight_reduction_sum_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight_reduction_sum_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii/model.mlir index 75c092767..043dd7234 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight_reduction_sum_ii(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight_reduction_sum_ii(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 0 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5,6,6],f32>, !torch.vtensor<[3,6,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii_expanded/model.mlir index 1c2fd4422..d68e23552 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2_with_weight_reduction_sum_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2_with_weight_reduction_sum_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6],f32>, %arg1: !torch.vtensor<[3,6,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii/model.mlir index 35d1f7fe5..86a42d75e 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3_none_no_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3_none_no_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.ignore_index = -5 : si64, torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6,5],f32>, !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> return %0 : !torch.vtensor<[3,6,6,5],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir index 685defc8b..f93b00323 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3_none_no_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3_none_no_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii/model.mlir index 664652df3..f446d9d59 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3_sum_weight_high_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3_sum_weight_high_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 10 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir index 455907ab2..9792d976a 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3_sum_weight_high_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3_sum_weight_high_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight/model.mlir index 76901a629..03f1aaecb 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3d4d5_mean_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3d4d5_mean_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight_expanded/model.mlir index 3e4fdea08..545d425c0 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3d4d5_mean_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3d4d5_mean_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight/model.mlir index f17597efd..9ee856bbe 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3d4d5_none_no_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3d4d5_none_no_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%arg0, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> return %0 : !torch.vtensor<[3,6,6,5,3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir index 6bdc391d6..4c986f83b 100644 --- a/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nllloss_NCd1d2d3d4d5_none_no_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nllloss_NCd1d2d3d4d5_none_no_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_center_point_box_format/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_center_point_box_format/model.mlir index a34f99f2c..464994b4e 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_center_point_box_format/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_center_point_box_format/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_center_point_box_format(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_center_point_box_format(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) {torch.onnx.center_point_box = 1 : si64} : (!torch.vtensor<[1,6,4],f32>, !torch.vtensor<[1,1,6],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> return %0 : !torch.vtensor<[3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_flipped_coordinates/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_flipped_coordinates/model.mlir index 184b8af6a..013d54b9c 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_flipped_coordinates/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_flipped_coordinates/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_flipped_coordinates(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_flipped_coordinates(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[1,6,4],f32>, !torch.vtensor<[1,1,6],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> return %0 : !torch.vtensor<[3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_identical_boxes/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_identical_boxes/model.mlir index 63f7890f2..a768b22ac 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_identical_boxes/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_identical_boxes/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_identical_boxes(%arg0: !torch.vtensor<[1,10,4],f32>, %arg1: !torch.vtensor<[1,1,10],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[1,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_identical_boxes(%arg0: !torch.vtensor<[1,10,4],f32>, %arg1: !torch.vtensor<[1,1,10],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[1,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[1,10,4],f32>, !torch.vtensor<[1,1,10],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[1,3],si64> return %0 : !torch.vtensor<[1,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_limit_output_size/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_limit_output_size/model.mlir index 273be8abc..838421e0f 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_limit_output_size/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_limit_output_size/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_limit_output_size(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[2,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_limit_output_size(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[2,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[1,6,4],f32>, !torch.vtensor<[1,1,6],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[2,3],si64> return %0 : !torch.vtensor<[2,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_single_box/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_single_box/model.mlir index c8c3949e6..025f47e83 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_single_box/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_single_box/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_single_box(%arg0: !torch.vtensor<[1,1,4],f32>, %arg1: !torch.vtensor<[1,1,1],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[1,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_single_box(%arg0: !torch.vtensor<[1,1,4],f32>, %arg1: !torch.vtensor<[1,1,1],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[1,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[1,1,4],f32>, !torch.vtensor<[1,1,1],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[1,3],si64> return %0 : !torch.vtensor<[1,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU/model.mlir index f82e18c60..9c07fc51b 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_suppress_by_IOU(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_suppress_by_IOU(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[1,6,4],f32>, !torch.vtensor<[1,1,6],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[3,3],si64> return %0 : !torch.vtensor<[3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU_and_scores/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU_and_scores/model.mlir index 05dcb5ddf..799473301 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU_and_scores/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_suppress_by_IOU_and_scores/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_suppress_by_IOU_and_scores(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[2,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_suppress_by_IOU_and_scores(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[2,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[1,6,4],f32>, !torch.vtensor<[1,1,6],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[2,3],si64> return %0 : !torch.vtensor<[2,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_batches/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_batches/model.mlir index 18904ddbf..f824cc65b 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_batches/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_batches/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_two_batches(%arg0: !torch.vtensor<[2,6,4],f32>, %arg1: !torch.vtensor<[2,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[4,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_two_batches(%arg0: !torch.vtensor<[2,6,4],f32>, %arg1: !torch.vtensor<[2,1,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[4,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[2,6,4],f32>, !torch.vtensor<[2,1,6],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[4,3],si64> return %0 : !torch.vtensor<[4,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_classes/model.mlir b/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_classes/model.mlir index b1f4f3262..ac58490a7 100644 --- a/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_classes/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonmaxsuppression_two_classes/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonmaxsuppression_two_classes(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,2,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[4,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonmaxsuppression_two_classes(%arg0: !torch.vtensor<[1,6,4],f32>, %arg1: !torch.vtensor<[1,2,6],f32>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],f32>, %arg4: !torch.vtensor<[1],f32>) -> !torch.vtensor<[4,3],si64> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonMaxSuppression"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[1,6,4],f32>, !torch.vtensor<[1,2,6],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],f32>) -> !torch.vtensor<[4,3],si64> return %0 : !torch.vtensor<[4,3],si64> diff --git a/iree_tests/onnx/node/generated/test_nonzero_example/model.mlir b/iree_tests/onnx/node/generated/test_nonzero_example/model.mlir index d9208405d..0800a4605 100644 --- a/iree_tests/onnx/node/generated/test_nonzero_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_nonzero_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_nonzero_example(%arg0: !torch.vtensor<[2,2],i1>) -> !torch.vtensor<[2,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_nonzero_example(%arg0: !torch.vtensor<[2,2],i1>) -> !torch.vtensor<[2,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.NonZero"(%arg0) : (!torch.vtensor<[2,2],i1>) -> !torch.vtensor<[2,3],si64> return %0 : !torch.vtensor<[2,3],si64> diff --git a/iree_tests/onnx/node/generated/test_not_2d/model.mlir b/iree_tests/onnx/node/generated/test_not_2d/model.mlir index 8eb05ec25..eceb9f83c 100644 --- a/iree_tests/onnx/node/generated/test_not_2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_not_2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_not_2d(%arg0: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_not_2d(%arg0: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Not"(%arg0) : (!torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> return %0 : !torch.vtensor<[3,4],i1> diff --git a/iree_tests/onnx/node/generated/test_not_3d/model.mlir b/iree_tests/onnx/node/generated/test_not_3d/model.mlir index 5eb03e127..501fcf6cd 100644 --- a/iree_tests/onnx/node/generated/test_not_3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_not_3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_not_3d(%arg0: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_not_3d(%arg0: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Not"(%arg0) : (!torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_not_4d/model.mlir b/iree_tests/onnx/node/generated/test_not_4d/model.mlir index 4b67dcd49..e30e9de9a 100644 --- a/iree_tests/onnx/node/generated/test_not_4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_not_4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_not_4d(%arg0: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_not_4d(%arg0: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Not"(%arg0) : (!torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_onehot_negative_indices/model.mlir b/iree_tests/onnx/node/generated/test_onehot_negative_indices/model.mlir index cec245eee..e93f7498f 100644 --- a/iree_tests/onnx/node/generated/test_onehot_negative_indices/model.mlir +++ b/iree_tests/onnx/node/generated/test_onehot_negative_indices/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_onehot_negative_indices(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[3,10],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_onehot_negative_indices(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[3,10],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.OneHot"(%arg0, %arg1, %arg2) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[3],si64>, !torch.vtensor<[],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[3,10],f32> return %0 : !torch.vtensor<[3,10],f32> diff --git a/iree_tests/onnx/node/generated/test_onehot_with_axis/model.mlir b/iree_tests/onnx/node/generated/test_onehot_with_axis/model.mlir index f7447936e..44d8ed154 100644 --- a/iree_tests/onnx/node/generated/test_onehot_with_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_onehot_with_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_onehot_with_axis(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2,10,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_onehot_with_axis(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2,10,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.OneHot"(%arg0, %arg1, %arg2) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[2,10,2],f32> return %0 : !torch.vtensor<[2,10,2],f32> diff --git a/iree_tests/onnx/node/generated/test_onehot_with_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_onehot_with_negative_axis/model.mlir index d1d7e2a7e..7aea6b662 100644 --- a/iree_tests/onnx/node/generated/test_onehot_with_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_onehot_with_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_onehot_with_negative_axis(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2,10,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_onehot_with_negative_axis(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2,10,2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.OneHot"(%arg0, %arg1, %arg2) {torch.onnx.axis = -2 : si64} : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[2,10,2],f32> return %0 : !torch.vtensor<[2,10,2],f32> diff --git a/iree_tests/onnx/node/generated/test_onehot_without_axis/model.mlir b/iree_tests/onnx/node/generated/test_onehot_without_axis/model.mlir index da2ccd1b1..dc3f984cf 100644 --- a/iree_tests/onnx/node/generated/test_onehot_without_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_onehot_without_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_onehot_without_axis(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],si32>) -> !torch.vtensor<[3,12],si32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_onehot_without_axis(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[2],si32>) -> !torch.vtensor<[3,12],si32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.OneHot"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],si64>, !torch.vtensor<[],f32>, !torch.vtensor<[2],si32>) -> !torch.vtensor<[3,12],si32> return %0 : !torch.vtensor<[3,12],si32> diff --git a/iree_tests/onnx/node/generated/test_or2d/model.mlir b/iree_tests/onnx/node/generated/test_or2d/model.mlir index 6f26dd73d..1ec1b5820 100644 --- a/iree_tests/onnx/node/generated/test_or2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or2d(%arg0: !torch.vtensor<[3,4],i1>, %arg1: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or2d(%arg0: !torch.vtensor<[3,4],i1>, %arg1: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[3,4],i1>, !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> return %0 : !torch.vtensor<[3,4],i1> diff --git a/iree_tests/onnx/node/generated/test_or3d/model.mlir b/iree_tests/onnx/node/generated/test_or3d/model.mlir index e5922d774..a459421ad 100644 --- a/iree_tests/onnx/node/generated/test_or3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or3d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or3d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_or4d/model.mlir b/iree_tests/onnx/node/generated/test_or4d/model.mlir index b5cc460a1..86e204adf 100644 --- a/iree_tests/onnx/node/generated/test_or4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or4d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or4d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_or_bcast3v1d/model.mlir b/iree_tests/onnx/node/generated/test_or_bcast3v1d/model.mlir index e2f3ef9d0..a7b9f8d1a 100644 --- a/iree_tests/onnx/node/generated/test_or_bcast3v1d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or_bcast3v1d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or_bcast3v1d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or_bcast3v1d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_or_bcast3v2d/model.mlir b/iree_tests/onnx/node/generated/test_or_bcast3v2d/model.mlir index e8e17070e..9d74d99ac 100644 --- a/iree_tests/onnx/node/generated/test_or_bcast3v2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or_bcast3v2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or_bcast3v2d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or_bcast3v2d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_or_bcast4v2d/model.mlir b/iree_tests/onnx/node/generated/test_or_bcast4v2d/model.mlir index 0f1b48ba9..f56ace3d5 100644 --- a/iree_tests/onnx/node/generated/test_or_bcast4v2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or_bcast4v2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or_bcast4v2d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or_bcast4v2d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_or_bcast4v3d/model.mlir b/iree_tests/onnx/node/generated/test_or_bcast4v3d/model.mlir index 86b1ec913..74e722111 100644 --- a/iree_tests/onnx/node/generated/test_or_bcast4v3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or_bcast4v3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or_bcast4v3d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or_bcast4v3d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_or_bcast4v4d/model.mlir b/iree_tests/onnx/node/generated/test_or_bcast4v4d/model.mlir index c1371aeb5..13d9576b1 100644 --- a/iree_tests/onnx/node/generated/test_or_bcast4v4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_or_bcast4v4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_or_bcast4v4d(%arg0: !torch.vtensor<[1,4,1,6],i1>, %arg1: !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_or_bcast4v4d(%arg0: !torch.vtensor<[1,4,1,6],i1>, %arg1: !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Or"(%arg0, %arg1) : (!torch.vtensor<[1,4,1,6],i1>, !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_pow/model.mlir b/iree_tests/onnx/node/generated/test_pow/model.mlir index 5b9267a33..2f774a11e 100644 --- a/iree_tests/onnx/node/generated/test_pow/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_bcast_array/model.mlir b/iree_tests/onnx/node/generated/test_pow_bcast_array/model.mlir index 7c2cbcf6d..73366b403 100644 --- a/iree_tests/onnx/node/generated/test_pow_bcast_array/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_bcast_array/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_bcast_array(%arg0: !torch.vtensor<[2,3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_bcast_array(%arg0: !torch.vtensor<[2,3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[2,3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[2,3],f32> return %0 : !torch.vtensor<[2,3],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_bcast_scalar/model.mlir b/iree_tests/onnx/node/generated/test_pow_bcast_scalar/model.mlir index 1ee5beb98..29136c1ce 100644 --- a/iree_tests/onnx/node/generated/test_pow_bcast_scalar/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_bcast_scalar/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_bcast_scalar(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_bcast_scalar(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_example/model.mlir b/iree_tests/onnx/node/generated/test_pow_example/model.mlir index a3dc669a3..71f455939 100644 --- a/iree_tests/onnx/node/generated/test_pow_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_types_float32_int32/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_float32_int32/model.mlir index 30e5517f9..bd021ba4b 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_float32_int32/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_float32_int32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_float32_int32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_float32_int32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_types_float32_int64/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_float32_int64/model.mlir index e242c1b71..c7169a89f 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_float32_int64/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_float32_int64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_float32_int64(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_float32_int64(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_types_float32_uint32/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_float32_uint32/model.mlir index a7e178d82..4cb2181de 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_float32_uint32/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_float32_uint32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_float32_uint32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_float32_uint32(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],ui32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_types_float32_uint64/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_float32_uint64/model.mlir index 4ab20a901..f65875d9b 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_float32_uint64/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_float32_uint64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_float32_uint64(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_float32_uint64(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],ui64>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_pow_types_int32_float32/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_int32_float32/model.mlir index b81e8e08d..8e782b51c 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_int32_float32/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_int32_float32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_int32_float32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_int32_float32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],si32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],si32> return %0 : !torch.vtensor<[3],si32> diff --git a/iree_tests/onnx/node/generated/test_pow_types_int32_int32/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_int32_int32/model.mlir index d192726b1..62bb4a9ff 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_int32_int32/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_int32_int32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_int32_int32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_int32_int32(%arg0: !torch.vtensor<[3],si32>, %arg1: !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],si32>, !torch.vtensor<[3],si32>) -> !torch.vtensor<[3],si32> return %0 : !torch.vtensor<[3],si32> diff --git a/iree_tests/onnx/node/generated/test_pow_types_int64_float32/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_int64_float32/model.mlir index 6b36d388e..15ec564db 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_int64_float32/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_int64_float32/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_int64_float32(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_int64_float32(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],si64>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],si64> return %0 : !torch.vtensor<[3],si64> diff --git a/iree_tests/onnx/node/generated/test_pow_types_int64_int64/model.mlir b/iree_tests/onnx/node/generated/test_pow_types_int64_int64/model.mlir index 594b5ae42..112005dd5 100644 --- a/iree_tests/onnx/node/generated/test_pow_types_int64_int64/model.mlir +++ b/iree_tests/onnx/node/generated/test_pow_types_int64_int64/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_pow_types_int64_int64(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 15 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_pow_types_int64_int64(%arg0: !torch.vtensor<[3],si64>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Pow"(%arg0, %arg1) : (!torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],si64> return %0 : !torch.vtensor<[3],si64> diff --git a/iree_tests/onnx/node/generated/test_prelu_broadcast/model.mlir b/iree_tests/onnx/node/generated/test_prelu_broadcast/model.mlir index 3f095203c..d8c832caa 100644 --- a/iree_tests/onnx/node/generated/test_prelu_broadcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_prelu_broadcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_prelu_broadcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_prelu_broadcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.PRelu"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_prelu_broadcast_expanded/model.mlir b/iree_tests/onnx/node/generated/test_prelu_broadcast_expanded/model.mlir index d54204555..45b7bd649 100644 --- a/iree_tests/onnx/node/generated/test_prelu_broadcast_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_prelu_broadcast_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_prelu_broadcast_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_prelu_broadcast_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0.000000e+00> : tensor} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.CastLike"(%0, %arg0) : (!torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_prelu_example/model.mlir b/iree_tests/onnx/node/generated/test_prelu_example/model.mlir index a843be50d..2bc482a8f 100644 --- a/iree_tests/onnx/node/generated/test_prelu_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_prelu_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_prelu_example(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_prelu_example(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.PRelu"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_prelu_example_expanded/model.mlir b/iree_tests/onnx/node/generated/test_prelu_example_expanded/model.mlir index 69d32f752..41ff92d40 100644 --- a/iree_tests/onnx/node/generated/test_prelu_example_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_prelu_example_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_prelu_example_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_prelu_example_expanded(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0.000000e+00> : tensor} : () -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.CastLike"(%0, %arg0) : (!torch.vtensor<[],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_qlinearconv/model.mlir b/iree_tests/onnx/node/generated/test_qlinearconv/model.mlir index f3b3e7eef..d73422624 100644 --- a/iree_tests/onnx/node/generated/test_qlinearconv/model.mlir +++ b/iree_tests/onnx/node/generated/test_qlinearconv/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_qlinearconv(%arg0: !torch.vtensor<[1,1,7,7],ui8>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],ui8>, %arg3: !torch.vtensor<[1,1,1,1],ui8>, %arg4: !torch.vtensor<[1],f32>, %arg5: !torch.vtensor<[1],ui8>, %arg6: !torch.vtensor<[],f32>, %arg7: !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,7,7],ui8> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_qlinearconv(%arg0: !torch.vtensor<[1,1,7,7],ui8>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],ui8>, %arg3: !torch.vtensor<[1,1,1,1],ui8>, %arg4: !torch.vtensor<[1],f32>, %arg5: !torch.vtensor<[1],ui8>, %arg6: !torch.vtensor<[],f32>, %arg7: !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,7,7],ui8> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.QLinearConv"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7) : (!torch.vtensor<[1,1,7,7],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>, !torch.vtensor<[1,1,1,1],ui8>, !torch.vtensor<[1],f32>, !torch.vtensor<[1],ui8>, !torch.vtensor<[],f32>, !torch.vtensor<[],ui8>) -> !torch.vtensor<[1,1,7,7],ui8> return %0 : !torch.vtensor<[1,1,7,7],ui8> diff --git a/iree_tests/onnx/node/generated/test_range_float_type_positive_delta/model.mlir b/iree_tests/onnx/node/generated/test_range_float_type_positive_delta/model.mlir index f1494eea3..6898969f6 100644 --- a/iree_tests/onnx/node/generated/test_range_float_type_positive_delta/model.mlir +++ b/iree_tests/onnx/node/generated/test_range_float_type_positive_delta/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_range_float_type_positive_delta(%arg0: !torch.vtensor<[],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_range_float_type_positive_delta(%arg0: !torch.vtensor<[],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Range"(%arg0, %arg1, %arg2) : (!torch.vtensor<[],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[2],f32> return %0 : !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_range_float_type_positive_delta_expanded/model.mlir b/iree_tests/onnx/node/generated/test_range_float_type_positive_delta_expanded/model.mlir index 06c9bf36b..a1c7eb30f 100644 --- a/iree_tests/onnx/node/generated/test_range_float_type_positive_delta_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_range_float_type_positive_delta_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_range_float_type_positive_delta_expanded(%arg0: !torch.vtensor<[],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_range_float_type_positive_delta_expanded(%arg0: !torch.vtensor<[],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sub"(%arg1, %arg0) : (!torch.vtensor<[],f32>, !torch.vtensor<[],f32>) -> !torch.vtensor<[],f32> %1 = torch.operator "onnx.Cast"(%0) {torch.onnx.to = 1 : si64} : (!torch.vtensor<[],f32>) -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta/model.mlir b/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta/model.mlir index eba535b11..770103137 100644 --- a/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta/model.mlir +++ b/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_range_int32_type_negative_delta(%arg0: !torch.vtensor<[],si32>, %arg1: !torch.vtensor<[],si32>, %arg2: !torch.vtensor<[],si32>) -> !torch.vtensor<[2],si32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_range_int32_type_negative_delta(%arg0: !torch.vtensor<[],si32>, %arg1: !torch.vtensor<[],si32>, %arg2: !torch.vtensor<[],si32>) -> !torch.vtensor<[2],si32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Range"(%arg0, %arg1, %arg2) : (!torch.vtensor<[],si32>, !torch.vtensor<[],si32>, !torch.vtensor<[],si32>) -> !torch.vtensor<[2],si32> return %0 : !torch.vtensor<[2],si32> diff --git a/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta_expanded/model.mlir b/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta_expanded/model.mlir index 53875ae47..3306bad83 100644 --- a/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_range_int32_type_negative_delta_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_range_int32_type_negative_delta_expanded(%arg0: !torch.vtensor<[],si32>, %arg1: !torch.vtensor<[],si32>, %arg2: !torch.vtensor<[],si32>) -> !torch.vtensor<[2],si32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_range_int32_type_negative_delta_expanded(%arg0: !torch.vtensor<[],si32>, %arg1: !torch.vtensor<[],si32>, %arg2: !torch.vtensor<[],si32>) -> !torch.vtensor<[2],si32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sub"(%arg1, %arg0) : (!torch.vtensor<[],si32>, !torch.vtensor<[],si32>) -> !torch.vtensor<[],si32> %1 = torch.operator "onnx.Cast"(%0) {torch.onnx.to = 1 : si64} : (!torch.vtensor<[],si32>) -> !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_reciprocal/model.mlir b/iree_tests/onnx/node/generated/test_reciprocal/model.mlir index 6edc50168..6ca709f9e 100644 --- a/iree_tests/onnx/node/generated/test_reciprocal/model.mlir +++ b/iree_tests/onnx/node/generated/test_reciprocal/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reciprocal(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reciprocal(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Reciprocal"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_reciprocal_example/model.mlir b/iree_tests/onnx/node/generated/test_reciprocal_example/model.mlir index 4b14cc65f..70ef52eda 100644 --- a/iree_tests/onnx/node/generated/test_reciprocal_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_reciprocal_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reciprocal_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reciprocal_example(%arg0: !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Reciprocal"(%arg0) : (!torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> return %0 : !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_example/model.mlir index 01d6f4495..77d531880 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_default_axes_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[1,1,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_default_axes_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[1,1,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[0],si64>) -> !torch.vtensor<[1,1,1],f32> return %0 : !torch.vtensor<[1,1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_random/model.mlir index 4c2f66e7e..db8d6cd9a 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_default_axes_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_default_axes_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[1,1,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_default_axes_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[1,1,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[0],si64>) -> !torch.vtensor<[1,1,1],f32> return %0 : !torch.vtensor<[1,1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_example/model.mlir index 447b0598a..31b331445 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_do_not_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_do_not_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 0 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,2],f32> return %0 : !torch.vtensor<[3,2],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_random/model.mlir index d4e1a8929..7d8c3f033 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_do_not_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_do_not_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_do_not_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 0 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,2],f32> return %0 : !torch.vtensor<[3,2],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_empty_axes_input_noop_example/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_empty_axes_input_noop_example/model.mlir index c26a1fe25..50d94d48f 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_empty_axes_input_noop_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_empty_axes_input_noop_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_empty_axes_input_noop_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[3,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_empty_axes_input_noop_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[3,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64, torch.onnx.noop_with_empty_axes = 1 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[0],si64>) -> !torch.vtensor<[3,2,2],f32> return %0 : !torch.vtensor<[3,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_empty_set/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_empty_set/model.mlir index ee6e78918..27a42a723 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_empty_set/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_empty_set/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_empty_set(%arg0: !torch.vtensor<[2,0,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[2,1,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_empty_set(%arg0: !torch.vtensor<[2,0,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[2,1,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,0,4],f32>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[2,1,4],f32> return %0 : !torch.vtensor<[2,1,4],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_empty_set_non_reduced_axis_zero/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_empty_set_non_reduced_axis_zero/model.mlir index efc27bd0a..be24c050d 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_empty_set_non_reduced_axis_zero/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_empty_set_non_reduced_axis_zero/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_empty_set_non_reduced_axis_zero(%arg0: !torch.vtensor<[2,0,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[2,0,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_empty_set_non_reduced_axis_zero(%arg0: !torch.vtensor<[2,0,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[2,0,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,0,4],f32>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[2,0,1],f32> return %0 : !torch.vtensor<[2,0,1],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_example/model.mlir index e1cf5e648..66515c34b 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> return %0 : !torch.vtensor<[3,1,2],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_random/model.mlir index 19524e565..acca772d0 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> return %0 : !torch.vtensor<[3,1,2],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_example/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_example/model.mlir index 6ac87dc6f..f20142732 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_negative_axes_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_negative_axes_keepdims_example(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[3,1,2],f32> return %0 : !torch.vtensor<[3,1,2],f32> diff --git a/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_random/model.mlir b/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_random/model.mlir index 462b2b732..acc1d88e7 100644 --- a/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_random/model.mlir +++ b/iree_tests/onnx/node/generated/test_reduce_sum_negative_axes_keepdims_random/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reduce_sum_negative_axes_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[3,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reduce_sum_negative_axes_keepdims_random(%arg0: !torch.vtensor<[3,2,2],f32>, %arg1: !torch.vtensor<[0],si64>) -> !torch.vtensor<[3,2,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReduceSum"(%arg0, %arg1) {torch.onnx.keepdims = 1 : si64, torch.onnx.noop_with_empty_axes = 1 : si64} : (!torch.vtensor<[3,2,2],f32>, !torch.vtensor<[0],si64>) -> !torch.vtensor<[3,2,2],f32> return %0 : !torch.vtensor<[3,2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_relu/model.mlir b/iree_tests/onnx/node/generated/test_relu/model.mlir index a406cc88c..6ff1e50d9 100644 --- a/iree_tests/onnx/node/generated/test_relu/model.mlir +++ b/iree_tests/onnx/node/generated/test_relu/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_relu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_relu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Relu"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_reversesequence_batch/model.mlir b/iree_tests/onnx/node/generated/test_reversesequence_batch/model.mlir index 0436989e3..8f9079d2d 100644 --- a/iree_tests/onnx/node/generated/test_reversesequence_batch/model.mlir +++ b/iree_tests/onnx/node/generated/test_reversesequence_batch/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reversesequence_batch(%arg0: !torch.vtensor<[4,4],f32>, %arg1: !torch.vtensor<[4],si64>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reversesequence_batch(%arg0: !torch.vtensor<[4,4],f32>, %arg1: !torch.vtensor<[4],si64>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReverseSequence"(%arg0, %arg1) {torch.onnx.batch_axis = 0 : si64, torch.onnx.time_axis = 1 : si64} : (!torch.vtensor<[4,4],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[4,4],f32> return %0 : !torch.vtensor<[4,4],f32> diff --git a/iree_tests/onnx/node/generated/test_reversesequence_time/model.mlir b/iree_tests/onnx/node/generated/test_reversesequence_time/model.mlir index 35f85b53a..54274cefe 100644 --- a/iree_tests/onnx/node/generated/test_reversesequence_time/model.mlir +++ b/iree_tests/onnx/node/generated/test_reversesequence_time/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_reversesequence_time(%arg0: !torch.vtensor<[4,4],f32>, %arg1: !torch.vtensor<[4],si64>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_reversesequence_time(%arg0: !torch.vtensor<[4,4],f32>, %arg1: !torch.vtensor<[4],si64>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ReverseSequence"(%arg0, %arg1) {torch.onnx.batch_axis = 1 : si64, torch.onnx.time_axis = 0 : si64} : (!torch.vtensor<[4,4],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[4,4],f32> return %0 : !torch.vtensor<[4,4],f32> diff --git a/iree_tests/onnx/node/generated/test_rnn_seq_length/model.mlir b/iree_tests/onnx/node/generated/test_rnn_seq_length/model.mlir index 133a37dca..bd7a5ca92 100644 --- a/iree_tests/onnx/node/generated/test_rnn_seq_length/model.mlir +++ b/iree_tests/onnx/node/generated/test_rnn_seq_length/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_rnn_seq_length(%arg0: !torch.vtensor<[2,3,3],f32>, %arg1: !torch.vtensor<[1,5,3],f32>, %arg2: !torch.vtensor<[1,5,5],f32>, %arg3: !torch.vtensor<[1,10],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_rnn_seq_length(%arg0: !torch.vtensor<[2,3,3],f32>, %arg1: !torch.vtensor<[1,5,3],f32>, %arg2: !torch.vtensor<[1,5,5],f32>, %arg3: !torch.vtensor<[1,10],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.RNN"(%arg0, %arg1, %arg2, %arg3) {torch.onnx.hidden_size = 5 : si64} : (!torch.vtensor<[2,3,3],f32>, !torch.vtensor<[1,5,3],f32>, !torch.vtensor<[1,5,5],f32>, !torch.vtensor<[1,10],f32>) -> (!torch.none, !torch.vtensor<[1,3,5],f32>) return %0#1 : !torch.vtensor<[1,3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_roialign_aligned_false/model.mlir b/iree_tests/onnx/node/generated/test_roialign_aligned_false/model.mlir index 593f960e0..64dd2bbb2 100644 --- a/iree_tests/onnx/node/generated/test_roialign_aligned_false/model.mlir +++ b/iree_tests/onnx/node/generated/test_roialign_aligned_false/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_roialign_aligned_false(%arg0: !torch.vtensor<[1,1,10,10],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_roialign_aligned_false(%arg0: !torch.vtensor<[1,1,10,10],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.RoiAlign"(%arg0, %arg1, %arg2) {torch.onnx.coordinate_transformation_mode = "output_half_pixel", torch.onnx.output_height = 5 : si64, torch.onnx.output_width = 5 : si64, torch.onnx.sampling_ratio = 2 : si64, torch.onnx.spatial_scale = 1.000000e+00 : f32} : (!torch.vtensor<[1,1,10,10],f32>, !torch.vtensor<[3,4],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> return %0 : !torch.vtensor<[3,1,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_roialign_aligned_true/model.mlir b/iree_tests/onnx/node/generated/test_roialign_aligned_true/model.mlir index 25a9171db..a85e98b8f 100644 --- a/iree_tests/onnx/node/generated/test_roialign_aligned_true/model.mlir +++ b/iree_tests/onnx/node/generated/test_roialign_aligned_true/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_roialign_aligned_true(%arg0: !torch.vtensor<[1,1,10,10],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_roialign_aligned_true(%arg0: !torch.vtensor<[1,1,10,10],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.RoiAlign"(%arg0, %arg1, %arg2) {torch.onnx.coordinate_transformation_mode = "half_pixel", torch.onnx.output_height = 5 : si64, torch.onnx.output_width = 5 : si64, torch.onnx.sampling_ratio = 2 : si64, torch.onnx.spatial_scale = 1.000000e+00 : f32} : (!torch.vtensor<[1,1,10,10],f32>, !torch.vtensor<[3,4],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> return %0 : !torch.vtensor<[3,1,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_roialign_mode_max/model.mlir b/iree_tests/onnx/node/generated/test_roialign_mode_max/model.mlir index 80683a152..d63a53359 100644 --- a/iree_tests/onnx/node/generated/test_roialign_mode_max/model.mlir +++ b/iree_tests/onnx/node/generated/test_roialign_mode_max/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_roialign_mode_max(%arg0: !torch.vtensor<[1,1,10,10],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_roialign_mode_max(%arg0: !torch.vtensor<[1,1,10,10],f32>, %arg1: !torch.vtensor<[3,4],f32>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.RoiAlign"(%arg0, %arg1, %arg2) {torch.onnx.coordinate_transformation_mode = "output_half_pixel", torch.onnx.mode = "max", torch.onnx.output_height = 5 : si64, torch.onnx.output_width = 5 : si64, torch.onnx.sampling_ratio = 2 : si64, torch.onnx.spatial_scale = 1.000000e+00 : f32} : (!torch.vtensor<[1,1,10,10],f32>, !torch.vtensor<[3,4],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,1,5,5],f32> return %0 : !torch.vtensor<[3,1,5,5],f32> diff --git a/iree_tests/onnx/node/generated/test_round/model.mlir b/iree_tests/onnx/node/generated/test_round/model.mlir index 30890342c..1a2821587 100644 --- a/iree_tests/onnx/node/generated/test_round/model.mlir +++ b/iree_tests/onnx/node/generated/test_round/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_round(%arg0: !torch.vtensor<[15],f32>) -> !torch.vtensor<[15],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_round(%arg0: !torch.vtensor<[15],f32>) -> !torch.vtensor<[15],f32> attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Round"(%arg0) : (!torch.vtensor<[15],f32>) -> !torch.vtensor<[15],f32> return %0 : !torch.vtensor<[15],f32> diff --git a/iree_tests/onnx/node/generated/test_scan9_sum/model.mlir b/iree_tests/onnx/node/generated/test_scan9_sum/model.mlir index 55ffc1c5f..d6521b598 100644 --- a/iree_tests/onnx/node/generated/test_scan9_sum/model.mlir +++ b/iree_tests/onnx/node/generated/test_scan9_sum/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_scan9_sum(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[3,2],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32>) attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_scan9_sum(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[3,2],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32>) attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Scan"(%arg0, %arg1) {torch.onnx.num_scan_inputs = 1 : si64} : (!torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32>) { ^bb0(%arg2: !torch.vtensor<[2],f32>, %arg3: !torch.vtensor<[2],f32>): diff --git a/iree_tests/onnx/node/generated/test_scan_sum/model.mlir b/iree_tests/onnx/node/generated/test_scan_sum/model.mlir index 27cb24b3f..d3d37e8ac 100644 --- a/iree_tests/onnx/node/generated/test_scan_sum/model.mlir +++ b/iree_tests/onnx/node/generated/test_scan_sum/model.mlir @@ -1,13 +1,13 @@ module { - func.func @test_scan_sum(%arg0: !torch.vtensor<[1,2],f32>, %arg1: !torch.vtensor<[1,3,2],f32>) -> (!torch.vtensor<[1,2],f32>, !torch.vtensor<[1,3,2],f32>) attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 8 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_scan_sum(%arg0: !torch.vtensor<[2],f32>, %arg1: !torch.vtensor<[3,2],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32>) attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none - %0:2 = torch.operator "onnx.Scan"(%none, %arg0, %arg1) {torch.onnx.num_scan_inputs = 1 : si64} : (!torch.none, !torch.vtensor<[1,2],f32>, !torch.vtensor<[1,3,2],f32>) -> (!torch.vtensor<[1,2],f32>, !torch.vtensor<[1,3,2],f32>) { + %0:2 = torch.operator "onnx.Scan"(%arg0, %arg1) {torch.onnx.num_scan_inputs = 1 : si64} : (!torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32>) { ^bb0(%arg2: !torch.vtensor<[2],f32>, %arg3: !torch.vtensor<[2],f32>): %1 = torch.operator "onnx.Add"(%arg2, %arg3) : (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> %2 = torch.operator "onnx.Identity"(%1) : (!torch.vtensor<[2],f32>) -> !torch.vtensor<[2],f32> torch.operator_terminator %1, %2 : !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32> } - return %0#0, %0#1 : !torch.vtensor<[1,2],f32>, !torch.vtensor<[1,3,2],f32> + return %0#0, %0#1 : !torch.vtensor<[2],f32>, !torch.vtensor<[3,2],f32> } } diff --git a/iree_tests/onnx/node/generated/test_scatter_with_axis/model.mlir b/iree_tests/onnx/node/generated/test_scatter_with_axis/model.mlir index 889171587..b0be6ce82 100644 --- a/iree_tests/onnx/node/generated/test_scatter_with_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_scatter_with_axis/model.mlir @@ -1,7 +1,7 @@ module { - func.func @test_scatter_with_axis(%arg0: !torch.vtensor<[1,5],f32>, %arg1: !torch.vtensor<[1,2],si64>, %arg2: !torch.vtensor<[1,2],f32>) -> !torch.vtensor<[1,5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_scatter_with_axis(%arg0: !torch.vtensor<[1,5],f32>, %arg1: !torch.vtensor<[1,2],si64>, %arg2: !torch.vtensor<[1,2],f32>) -> !torch.vtensor<[1,5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none - %0 = torch.operator "onnx.Scatter"(%arg0, %arg1, %arg2) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[1,5],f32>, !torch.vtensor<[1,2],si64>, !torch.vtensor<[1,2],f32>) -> !torch.vtensor<[1,5],f32> + %0 = torch.operator "onnx.ScatterElements"(%arg0, %arg1, %arg2) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[1,5],f32>, !torch.vtensor<[1,2],si64>, !torch.vtensor<[1,2],f32>) -> !torch.vtensor<[1,5],f32> return %0 : !torch.vtensor<[1,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_scatter_without_axis/model.mlir b/iree_tests/onnx/node/generated/test_scatter_without_axis/model.mlir index 8a9afc927..51a408b9f 100644 --- a/iree_tests/onnx/node/generated/test_scatter_without_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_scatter_without_axis/model.mlir @@ -1,7 +1,7 @@ module { - func.func @test_scatter_without_axis(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[2,3],si64>, %arg2: !torch.vtensor<[2,3],f32>) -> !torch.vtensor<[3,3],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_scatter_without_axis(%arg0: !torch.vtensor<[3,3],f32>, %arg1: !torch.vtensor<[2,3],si64>, %arg2: !torch.vtensor<[2,3],f32>) -> !torch.vtensor<[3,3],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none - %0 = torch.operator "onnx.Scatter"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3,3],f32>, !torch.vtensor<[2,3],si64>, !torch.vtensor<[2,3],f32>) -> !torch.vtensor<[3,3],f32> + %0 = torch.operator "onnx.ScatterElements"(%arg0, %arg1, %arg2) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[3,3],f32>, !torch.vtensor<[2,3],si64>, !torch.vtensor<[2,3],f32>) -> !torch.vtensor<[3,3],f32> return %0 : !torch.vtensor<[3,3],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii/model.mlir index 273e62bec..5ac4c63d3 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1_mean_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1_mean_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = -1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded/model.mlir index fad236a2d..4b36e75ab 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1_mean_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1_mean_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,6],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,6,5],f32>) -> !torch.vtensor<[3,6,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,6,5],f32>) -> !torch.vtensor<[3,5,6],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = -1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = -1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob/model.mlir index f3d86d1b3..025842dac 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1_mean_weight_negative_ii_log_prob(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1_mean_weight_negative_ii_log_prob(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = -1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3,6],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,6],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob_expanded/model.mlir index 9d51cccdb..ee90b011a 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1_mean_weight_negative_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1_mean_weight_negative_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6],f32>, %arg1: !torch.vtensor<[3,6],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,6],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,6,5],f32>) -> !torch.vtensor<[3,6,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,6,5],f32>) -> !torch.vtensor<[3,5,6],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,6],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = -1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,6],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?],f32>) -> !torch.vtensor<[3,5,6],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = -1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,6],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,6],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii/model.mlir index 39507d1be..9db140d1c 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = -5 : si64, torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6,5],f32>, !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> return %0 : !torch.vtensor<[3,6,6,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir index 944b3d755..c823f6073 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6,6,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,180],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,180,5],f32>) -> !torch.vtensor<[3,180,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,180,5],f32>) -> !torch.vtensor<[3,5,180],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6,6,5],f32>) -> !torch.vtensor<[5],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,180],f32>, !torch.vtensor<[5],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = -5 : si64, torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,180],f32>, !torch.vtensor<[5],si64>) -> !torch.vtensor<[?,?,?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = -5 : si64, torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> return %7 : !torch.vtensor<[3,6,6,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob/model.mlir index 96cca2bea..4b4b91937 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> (!torch.vtensor<[3,6,6,5],f32>, !torch.vtensor<[3,5,6,6,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> (!torch.vtensor<[3,6,6,5],f32>, !torch.vtensor<[3,5,6,6,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = -5 : si64, torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6,5],f32>, !torch.vtensor<[3,6,6,5],si64>) -> (!torch.vtensor<[3,6,6,5],f32>, !torch.vtensor<[3,5,6,6,5],f32>) return %0#0, %0#1 : !torch.vtensor<[3,6,6,5],f32>, !torch.vtensor<[3,5,6,6,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded/model.mlir index bbe594621..d1ee8d7a1 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> (!torch.vtensor<[3,6,6,5],f32>, !torch.vtensor<[3,5,6,6,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6,6,5],f32>, %arg1: !torch.vtensor<[3,6,6,5],si64>) -> (!torch.vtensor<[3,6,6,5],f32>, !torch.vtensor<[3,5,6,6,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6,6,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,180],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,180,5],f32>) -> !torch.vtensor<[3,180,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,180,5],f32>) -> !torch.vtensor<[3,5,180],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6,6,5],f32>) -> !torch.vtensor<[5],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,180],f32>, !torch.vtensor<[5],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,6,6,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = -5 : si64, torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,180],f32>, !torch.vtensor<[5],si64>) -> !torch.vtensor<[?,?,?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?,?,?],f32>) -> !torch.vtensor<[3,5,6,6,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = -5 : si64, torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?,?,?,?],f32>, !torch.vtensor<[3,6,6,5],si64>) -> !torch.vtensor<[3,6,6,5],f32> return %8, %7 : !torch.vtensor<[3,6,6,5],f32>, !torch.vtensor<[3,5,6,6,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii/model.mlir index 7673851e5..dbf0cf85d 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_sum_weight_high_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_sum_weight_high_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 10 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir index 1301e1b95..49e3e596e 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_sum_weight_high_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_sum_weight_high_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 10 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 10 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob/model.mlir index afa0b61b0..378bf67c5 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_sum_weight_high_ii_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_sum_weight_high_ii_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 10 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded/model.mlir index 3372e61bb..3f4051e5c 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 10 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 10 : si64, torch.onnx.reduction = "sum"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight/model.mlir index e32ac302e..17093593b 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_mean_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_mean_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_expanded/model.mlir index c9ceee159..2fbe9d1b9 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_mean_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_mean_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2160],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,2160,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,5,2160],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>) -> !torch.vtensor<[7],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[?,?,?,?,?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?,?,?,?,?],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob/model.mlir index a63a8914c..81ebf1047 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_mean_weight_log_prob(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_mean_weight_log_prob(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob_expanded/model.mlir index cc7cc9c36..416d8a00a 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_mean_weight_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_mean_weight_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2160],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,2160,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,5,2160],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>) -> !torch.vtensor<[7],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,6,6,5,3,4],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[?,?,?,?,?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?,?,?,?,?],f32>) -> !torch.vtensor<[3,5,6,6,5,3,4],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?,?,?,?,?],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight/model.mlir index 0c8158da3..e786ee740 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_none_no_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_none_no_weight(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> return %0 : !torch.vtensor<[3,6,6,5,3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir index 86a9eeabd..b0901eb9a 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_none_no_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_none_no_weight_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2160],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,2160,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,5,2160],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>) -> !torch.vtensor<[7],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[?,?,?,?,?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?,?,?,?,?,?],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> return %7 : !torch.vtensor<[3,6,6,5,3,4],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob/model.mlir index 87d2d0ded..fbaba79e5 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_none_no_weight_log_prob(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> (!torch.vtensor<[3,6,6,5,3,4],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_none_no_weight_log_prob(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> (!torch.vtensor<[3,6,6,5,3,4],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>) -> (!torch.vtensor<[3,6,6,5,3,4],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) return %0#0, %0#1 : !torch.vtensor<[3,6,6,5,3,4],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob_expanded/model.mlir index 3c6c3e79a..fd899d1d2 100644 --- a/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_NCd1d2d3d4d5_none_no_weight_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_NCd1d2d3d4d5_none_no_weight_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> (!torch.vtensor<[3,6,6,5,3,4],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_NCd1d2d3d4d5_none_no_weight_log_prob_expanded(%arg0: !torch.vtensor<[3,5,6,6,5,3,4],f32>, %arg1: !torch.vtensor<[3,6,6,5,3,4],si64>) -> (!torch.vtensor<[3,6,6,5,3,4],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2160],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,2160,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2160,5],f32>) -> !torch.vtensor<[3,5,2160],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,6,6,5,3,4],f32>) -> !torch.vtensor<[7],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,6,6,5,3,4],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2160],f32>, !torch.vtensor<[7],si64>) -> !torch.vtensor<[?,?,?,?,?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?,?,?,?,?],f32>) -> !torch.vtensor<[3,5,6,6,5,3,4],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?,?,?,?,?,?],f32>, !torch.vtensor<[3,6,6,5,3,4],si64>) -> !torch.vtensor<[3,6,6,5,3,4],f32> return %8, %7 : !torch.vtensor<[3,6,6,5,3,4],f32>, !torch.vtensor<[3,5,6,6,5,3,4],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean/model.mlir index 6e85a410f..1e287090a 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_3d/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_3d/model.mlir index f75287b73..1a183c4e0 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_3d(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_3d(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_3d_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_3d_expanded/model.mlir index 89cf19e51..628ba1c0a 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_3d_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_3d_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_3d_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_3d_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,2,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,5,2],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob/model.mlir index 28e78f823..dbac4ed9c 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_3d_log_prob(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_3d_log_prob(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob_expanded/model.mlir index cfc174544..af4251bf8 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_3d_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_3d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_3d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,2,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,5,2],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,2],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?],f32>) -> !torch.vtensor<[3,5,2],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_expanded/model.mlir index 1fc2a48fc..a83cc8f85 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_log_prob/model.mlir index 3efc74a92..a9046ec1d 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_log_prob_expanded/model.mlir index c9d83dbab..a6c244317 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii/model.mlir index e66e26904..6e47681d5 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d/model.mlir index 52dee987a..8b0be7d79 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_3d(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_3d(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_expanded/model.mlir index 99f1fa23e..c386761f1 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_3d_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_3d_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,2,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,5,2],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob/model.mlir index f900da20a..bec1afabd 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_3d_log_prob(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_3d_log_prob(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob_expanded/model.mlir index b6e5abfcb..389d7ade9 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_3d_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_3d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_3d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,2,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,5,2],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,2],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?],f32>) -> !torch.vtensor<[3,5,2],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,2],si64>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d/model.mlir index 2ac3dd153..242ba6eb8 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_4d(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_4d(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_expanded/model.mlir index 736d6b593..41b3609aa 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_4d_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_4d_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,14],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,14,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,5,14],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2,7],f32>) -> !torch.vtensor<[4],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[?,?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?,?],f32>, !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob/model.mlir index 5662787df..14081da57 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_4d_log_prob(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_4d_log_prob(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3,2,7],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob_expanded/model.mlir index 0eabd1e7e..36777c352 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_4d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_4d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,14],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,14,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,5,14],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2,7],f32>) -> !torch.vtensor<[4],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,2,7],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[?,?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?,?],f32>) -> !torch.vtensor<[3,5,2,7],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?,?],f32>, !torch.vtensor<[3,2,7],si64>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_expanded/model.mlir index ba1c59f7b..3941b9348 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob/model.mlir index ac5d0a562..ab81219ba 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded/model.mlir index a9ba17b8b..92deb6ae7 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_no_weight_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_no_weight_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight/model.mlir index 70cef1d7a..cb6ab5163 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_expanded/model.mlir index 04c2bfc1f..a6a170e94 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii/model.mlir index 859b2ca1a..d205a96d7 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 0 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d/model.mlir index 73e6bdb60..2c133edbc 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_3d(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_3d(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_expanded/model.mlir index 59d0621b3..60d507596 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_3d_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_3d_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,2,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,5,2],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob/model.mlir index 67ebcd0a6..b48611594 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_3d_log_prob(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_3d_log_prob(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob_expanded/model.mlir index 2da661f95..ad3ca3dab 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_3d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_3d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2],f32>, %arg1: !torch.vtensor<[3,2],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,2],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,2,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,2,5],f32>) -> !torch.vtensor<[3,5,2],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2],f32>) -> !torch.vtensor<[3],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,2],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,2],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?],f32>) -> !torch.vtensor<[3,5,2],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 1 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?],f32>, !torch.vtensor<[3,2],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d/model.mlir index 685b481a2..d131860fb 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_4d(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_4d(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3,2,7],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_expanded/model.mlir index 45fd9296e..a90d45982 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_4d_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_4d_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,14],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,14,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,5,14],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2,7],f32>) -> !torch.vtensor<[4],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2,7],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[?,?,?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?,?],f32>, !torch.vtensor<[3,2,7],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob/model.mlir index 2304386ab..ff2c0387c 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_4d_log_prob(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_4d_log_prob(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3,2,7],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob_expanded/model.mlir index a76df3959..88fb63cc3 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_4d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_4d_log_prob_expanded(%arg0: !torch.vtensor<[3,5,2,7],f32>, %arg1: !torch.vtensor<[3,2,7],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5,2,7],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,14],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,14,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,14,5],f32>) -> !torch.vtensor<[3,5,14],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5,2,7],f32>) -> !torch.vtensor<[4],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5,2,7],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3,2,7],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,14],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[?,?,?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?,?,?],f32>) -> !torch.vtensor<[3,5,2,7],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 2 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?,?,?],f32>, !torch.vtensor<[3,2,7],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5,2,7],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_expanded/model.mlir index 25bd0f8d8..5730c72d9 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 0 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 0 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob/model.mlir index 7a1bbe076..f2b1677fa 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.ignore_index = 0 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob_expanded/model.mlir index 0a7585f66..e5fd9b2e1 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_ii_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_ii_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 0 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.ignore_index = 0 : si64, torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob/model.mlir index a392b710c..638ad75e9 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob_expanded/model.mlir index 2aca0f9a5..08d6b4638 100644 --- a/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_mean_weight_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_mean_weight_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_mean_weight_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "mean"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_none/model.mlir b/iree_tests/onnx/node/generated/test_sce_none/model.mlir index fb86a5de7..fbed10e23 100644 --- a/iree_tests/onnx/node/generated/test_sce_none/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_none_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_none_expanded/model.mlir index ab6f4e474..238f367ad 100644 --- a/iree_tests/onnx/node/generated/test_sce_none_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> return %7 : !torch.vtensor<[3],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_none_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_none_log_prob/model.mlir index 330a4da35..d79dc8afd 100644 --- a/iree_tests/onnx/node/generated/test_sce_none_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_none_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_none_log_prob_expanded/model.mlir index 0213d1f6a..8ea5da4ed 100644 --- a/iree_tests/onnx/node/generated/test_sce_none_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3],f32> return %8, %7 : !torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_none_weights/model.mlir b/iree_tests/onnx/node/generated/test_sce_none_weights/model.mlir index 0682b70c0..d841e7421 100644 --- a/iree_tests/onnx/node/generated/test_sce_none_weights/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none_weights/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none_weights(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none_weights(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_none_weights_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_none_weights_expanded/model.mlir index eef7c6252..ce900ac8c 100644 --- a/iree_tests/onnx/node/generated/test_sce_none_weights_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none_weights_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none_weights_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none_weights_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> return %7 : !torch.vtensor<[3],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob/model.mlir index 061cc66d6..429aeeb35 100644 --- a/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none_weights_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none_weights_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1, %arg2) {torch.onnx.reduction = "none"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob_expanded/model.mlir index 14a84cc6b..8e726d747 100644 --- a/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_none_weights_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_none_weights_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_none_weights_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[5],f32>) -> (!torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "none"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1, %arg2) {torch.onnx.reduction = "none"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3],f32> return %8, %7 : !torch.vtensor<[3],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_sum/model.mlir b/iree_tests/onnx/node/generated/test_sce_sum/model.mlir index d8e558d86..13993d668 100644 --- a/iree_tests/onnx/node/generated/test_sce_sum/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_sum/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_sum(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_sum(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %0 : !torch.vtensor<[],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_sum_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_sum_expanded/model.mlir index b107fbad3..7604ba4e0 100644 --- a/iree_tests/onnx/node/generated/test_sce_sum_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_sum_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_sum_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_sum_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,8 +7,8 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %7 : !torch.vtensor<[],f32> } } diff --git a/iree_tests/onnx/node/generated/test_sce_sum_log_prob/model.mlir b/iree_tests/onnx/node/generated/test_sce_sum_log_prob/model.mlir index f17934081..0052095ee 100644 --- a/iree_tests/onnx/node/generated/test_sce_sum_log_prob/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_sum_log_prob/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_sum_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_sum_log_prob(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.SoftmaxCrossEntropyLoss"(%arg0, %arg1) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) return %0#0, %0#1 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sce_sum_log_prob_expanded/model.mlir b/iree_tests/onnx/node/generated/test_sce_sum_log_prob_expanded/model.mlir index b2f41e87b..23c547841 100644 --- a/iree_tests/onnx/node/generated/test_sce_sum_log_prob_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_sce_sum_log_prob_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sce_sum_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sce_sum_log_prob_expanded(%arg0: !torch.vtensor<[3,5],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0, 0, -1]> : tensor<3xsi64>} : () -> !torch.vtensor<[3],si64> %1 = torch.operator "onnx.Reshape"(%arg0, %0) : (!torch.vtensor<[3,5],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[3,5,1],f32> @@ -7,9 +7,9 @@ module { %3 = torch.operator "onnx.LogSoftmax"(%2) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,1,5],f32> %4 = torch.operator "onnx.Transpose"(%3) {torch.onnx.perm = [0 : si64, 2 : si64, 1 : si64]} : (!torch.vtensor<[3,1,5],f32>) -> !torch.vtensor<[3,5,1],f32> %5 = torch.operator "onnx.Shape"(%arg0) : (!torch.vtensor<[3,5],f32>) -> !torch.vtensor<[2],si64> - %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[],f32> - %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[],f32>) -> !torch.vtensor<[3,5],f32> - %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> + %6 = torch.operator "onnx.Reshape"(%4, %5) : (!torch.vtensor<[3,5,1],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[?,?],f32> + %7 = torch.operator "onnx.Identity"(%6) : (!torch.vtensor<[?,?],f32>) -> !torch.vtensor<[3,5],f32> + %8 = torch.operator "onnx.NegativeLogLikelihoodLoss"(%6, %arg1) {torch.onnx.reduction = "sum"} : (!torch.vtensor<[?,?],f32>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[],f32> return %8, %7 : !torch.vtensor<[],f32>, !torch.vtensor<[3,5],f32> } } diff --git a/iree_tests/onnx/node/generated/test_selu/model.mlir b/iree_tests/onnx/node/generated/test_selu/model.mlir index be2b33de8..7950ca92d 100644 --- a/iree_tests/onnx/node/generated/test_selu/model.mlir +++ b/iree_tests/onnx/node/generated/test_selu/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_selu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_selu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Selu"(%arg0) {torch.onnx.alpha = 2.000000e+00 : f32, torch.onnx.gamma = 3.000000e+00 : f32} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_selu_default/model.mlir b/iree_tests/onnx/node/generated/test_selu_default/model.mlir index b3ebb5e15..1b7a79b3b 100644 --- a/iree_tests/onnx/node/generated/test_selu_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_selu_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_selu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_selu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Selu"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_selu_example/model.mlir b/iree_tests/onnx/node/generated/test_selu_example/model.mlir index 12d094772..b25b7a332 100644 --- a/iree_tests/onnx/node/generated/test_selu_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_selu_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_selu_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 6 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_selu_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Selu"(%arg0) {torch.onnx.alpha = 2.000000e+00 : f32, torch.onnx.gamma = 3.000000e+00 : f32} : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_shrink_hard/model.mlir b/iree_tests/onnx/node/generated/test_shrink_hard/model.mlir index d6320bc15..1dc1e4773 100644 --- a/iree_tests/onnx/node/generated/test_shrink_hard/model.mlir +++ b/iree_tests/onnx/node/generated/test_shrink_hard/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_shrink_hard(%arg0: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_shrink_hard(%arg0: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Shrink"(%arg0) {torch.onnx.lambd = 1.500000e+00 : f32} : (!torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> return %0 : !torch.vtensor<[5],f32> diff --git a/iree_tests/onnx/node/generated/test_shrink_soft/model.mlir b/iree_tests/onnx/node/generated/test_shrink_soft/model.mlir index 33cd54a0a..e44b4bba5 100644 --- a/iree_tests/onnx/node/generated/test_shrink_soft/model.mlir +++ b/iree_tests/onnx/node/generated/test_shrink_soft/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_shrink_soft(%arg0: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_shrink_soft(%arg0: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Shrink"(%arg0) {torch.onnx.bias = 1.500000e+00 : f32, torch.onnx.lambd = 1.500000e+00 : f32} : (!torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> return %0 : !torch.vtensor<[5],f32> diff --git a/iree_tests/onnx/node/generated/test_sigmoid/model.mlir b/iree_tests/onnx/node/generated/test_sigmoid/model.mlir index 19c6637cc..62c3ccbfa 100644 --- a/iree_tests/onnx/node/generated/test_sigmoid/model.mlir +++ b/iree_tests/onnx/node/generated/test_sigmoid/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sigmoid(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sigmoid(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sigmoid"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sigmoid_example/model.mlir b/iree_tests/onnx/node/generated/test_sigmoid_example/model.mlir index 5bbadbf87..ef76e286f 100644 --- a/iree_tests/onnx/node/generated/test_sigmoid_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_sigmoid_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sigmoid_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sigmoid_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sigmoid"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_sign/model.mlir b/iree_tests/onnx/node/generated/test_sign/model.mlir index e23d9d14c..c25517a48 100644 --- a/iree_tests/onnx/node/generated/test_sign/model.mlir +++ b/iree_tests/onnx/node/generated/test_sign/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sign(%arg0: !torch.vtensor<[11],f32>) -> !torch.vtensor<[11],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sign(%arg0: !torch.vtensor<[11],f32>) -> !torch.vtensor<[11],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sign"(%arg0) : (!torch.vtensor<[11],f32>) -> !torch.vtensor<[11],f32> return %0 : !torch.vtensor<[11],f32> diff --git a/iree_tests/onnx/node/generated/test_simple_rnn_batchwise/model.mlir b/iree_tests/onnx/node/generated/test_simple_rnn_batchwise/model.mlir index 6f417f19a..b8fffe4d0 100644 --- a/iree_tests/onnx/node/generated/test_simple_rnn_batchwise/model.mlir +++ b/iree_tests/onnx/node/generated/test_simple_rnn_batchwise/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_simple_rnn_batchwise(%arg0: !torch.vtensor<[3,1,2],f32>, %arg1: !torch.vtensor<[1,4,2],f32>, %arg2: !torch.vtensor<[1,4,4],f32>) -> (!torch.vtensor<[3,1,1,4],f32>, !torch.vtensor<[3,1,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_simple_rnn_batchwise(%arg0: !torch.vtensor<[3,1,2],f32>, %arg1: !torch.vtensor<[1,4,2],f32>, %arg2: !torch.vtensor<[1,4,4],f32>) -> (!torch.vtensor<[3,1,1,4],f32>, !torch.vtensor<[3,1,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.RNN"(%arg0, %arg1, %arg2) {torch.onnx.hidden_size = 4 : si64, torch.onnx.layout = 1 : si64} : (!torch.vtensor<[3,1,2],f32>, !torch.vtensor<[1,4,2],f32>, !torch.vtensor<[1,4,4],f32>) -> (!torch.vtensor<[3,1,1,4],f32>, !torch.vtensor<[3,1,4],f32>) return %0#0, %0#1 : !torch.vtensor<[3,1,1,4],f32>, !torch.vtensor<[3,1,4],f32> diff --git a/iree_tests/onnx/node/generated/test_simple_rnn_defaults/model.mlir b/iree_tests/onnx/node/generated/test_simple_rnn_defaults/model.mlir index a852f56f7..6aa6063ce 100644 --- a/iree_tests/onnx/node/generated/test_simple_rnn_defaults/model.mlir +++ b/iree_tests/onnx/node/generated/test_simple_rnn_defaults/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_simple_rnn_defaults(%arg0: !torch.vtensor<[1,3,2],f32>, %arg1: !torch.vtensor<[1,4,2],f32>, %arg2: !torch.vtensor<[1,4,4],f32>) -> !torch.vtensor<[1,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_simple_rnn_defaults(%arg0: !torch.vtensor<[1,3,2],f32>, %arg1: !torch.vtensor<[1,4,2],f32>, %arg2: !torch.vtensor<[1,4,4],f32>) -> !torch.vtensor<[1,3,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.RNN"(%arg0, %arg1, %arg2) {torch.onnx.hidden_size = 4 : si64} : (!torch.vtensor<[1,3,2],f32>, !torch.vtensor<[1,4,2],f32>, !torch.vtensor<[1,4,4],f32>) -> (!torch.none, !torch.vtensor<[1,3,4],f32>) return %0#1 : !torch.vtensor<[1,3,4],f32> diff --git a/iree_tests/onnx/node/generated/test_simple_rnn_with_initial_bias/model.mlir b/iree_tests/onnx/node/generated/test_simple_rnn_with_initial_bias/model.mlir index 76b3e7d76..240b113cc 100644 --- a/iree_tests/onnx/node/generated/test_simple_rnn_with_initial_bias/model.mlir +++ b/iree_tests/onnx/node/generated/test_simple_rnn_with_initial_bias/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_simple_rnn_with_initial_bias(%arg0: !torch.vtensor<[1,3,3],f32>, %arg1: !torch.vtensor<[1,5,3],f32>, %arg2: !torch.vtensor<[1,5,5],f32>, %arg3: !torch.vtensor<[1,10],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_simple_rnn_with_initial_bias(%arg0: !torch.vtensor<[1,3,3],f32>, %arg1: !torch.vtensor<[1,5,3],f32>, %arg2: !torch.vtensor<[1,5,5],f32>, %arg3: !torch.vtensor<[1,10],f32>) -> !torch.vtensor<[1,3,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.RNN"(%arg0, %arg1, %arg2, %arg3) {torch.onnx.hidden_size = 5 : si64} : (!torch.vtensor<[1,3,3],f32>, !torch.vtensor<[1,5,3],f32>, !torch.vtensor<[1,5,5],f32>, !torch.vtensor<[1,10],f32>) -> (!torch.none, !torch.vtensor<[1,3,5],f32>) return %0#1 : !torch.vtensor<[1,3,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sin/model.mlir b/iree_tests/onnx/node/generated/test_sin/model.mlir index 928aa896b..e110b33e6 100644 --- a/iree_tests/onnx/node/generated/test_sin/model.mlir +++ b/iree_tests/onnx/node/generated/test_sin/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sin(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sin(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sin"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sin_example/model.mlir b/iree_tests/onnx/node/generated/test_sin_example/model.mlir index e635d3808..b37310be7 100644 --- a/iree_tests/onnx/node/generated/test_sin_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_sin_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sin_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sin_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sin"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_sinh/model.mlir b/iree_tests/onnx/node/generated/test_sinh/model.mlir index 3b5e11d4f..5aeb79bbc 100644 --- a/iree_tests/onnx/node/generated/test_sinh/model.mlir +++ b/iree_tests/onnx/node/generated/test_sinh/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sinh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sinh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sinh"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sinh_example/model.mlir b/iree_tests/onnx/node/generated/test_sinh_example/model.mlir index 2e6bd657c..671e86d46 100644 --- a/iree_tests/onnx/node/generated/test_sinh_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_sinh_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sinh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sinh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sinh"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_slice/model.mlir b/iree_tests/onnx/node/generated/test_slice/model.mlir index 5188678cd..b166887c0 100644 --- a/iree_tests/onnx/node/generated/test_slice/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[2],si64>, %arg2: !torch.vtensor<[2],si64>, %arg3: !torch.vtensor<[2],si64>, %arg4: !torch.vtensor<[2],si64>) -> !torch.vtensor<[3,10,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[2],si64>, %arg2: !torch.vtensor<[2],si64>, %arg3: !torch.vtensor<[2],si64>, %arg4: !torch.vtensor<[2],si64>) -> !torch.vtensor<[3,10,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[2],si64>, !torch.vtensor<[2],si64>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[3,10,5],f32> return %0 : !torch.vtensor<[3,10,5],f32> diff --git a/iree_tests/onnx/node/generated/test_slice_default_axes/model.mlir b/iree_tests/onnx/node/generated/test_slice_default_axes/model.mlir index 4ddaf07a2..e3c2449bc 100644 --- a/iree_tests/onnx/node/generated/test_slice_default_axes/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice_default_axes/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice_default_axes(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice_default_axes(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> return %0 : !torch.vtensor<[20,10,1],f32> diff --git a/iree_tests/onnx/node/generated/test_slice_default_steps/model.mlir b/iree_tests/onnx/node/generated/test_slice_default_steps/model.mlir index daf02110a..32ff64492 100644 --- a/iree_tests/onnx/node/generated/test_slice_default_steps/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice_default_steps/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice_default_steps(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>, %arg3: !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice_default_steps(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>, %arg3: !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2, %arg3) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> return %0 : !torch.vtensor<[20,10,1],f32> diff --git a/iree_tests/onnx/node/generated/test_slice_end_out_of_bounds/model.mlir b/iree_tests/onnx/node/generated/test_slice_end_out_of_bounds/model.mlir index 64fe48e66..f072d4113 100644 --- a/iree_tests/onnx/node/generated/test_slice_end_out_of_bounds/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice_end_out_of_bounds/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice_end_out_of_bounds(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[1],si64>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],si64>, %arg4: !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,9,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice_end_out_of_bounds(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[1],si64>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],si64>, %arg4: !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,9,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,9,5],f32> return %0 : !torch.vtensor<[20,9,5],f32> diff --git a/iree_tests/onnx/node/generated/test_slice_neg/model.mlir b/iree_tests/onnx/node/generated/test_slice_neg/model.mlir index 7b43d4e3e..a9dcdc169 100644 --- a/iree_tests/onnx/node/generated/test_slice_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice_neg(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[1],si64>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],si64>, %arg4: !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,9,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice_neg(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[1],si64>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],si64>, %arg4: !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,9,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,9,5],f32> return %0 : !torch.vtensor<[20,9,5],f32> diff --git a/iree_tests/onnx/node/generated/test_slice_neg_steps/model.mlir b/iree_tests/onnx/node/generated/test_slice_neg_steps/model.mlir index 0c2bd6629..9c982544f 100644 --- a/iree_tests/onnx/node/generated/test_slice_neg_steps/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice_neg_steps/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice_neg_steps(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>, %arg3: !torch.vtensor<[3],si64>, %arg4: !torch.vtensor<[3],si64>) -> !torch.vtensor<[19,3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice_neg_steps(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>, %arg3: !torch.vtensor<[3],si64>, %arg4: !torch.vtensor<[3],si64>) -> !torch.vtensor<[19,3,2],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[19,3,2],f32> return %0 : !torch.vtensor<[19,3,2],f32> diff --git a/iree_tests/onnx/node/generated/test_slice_negative_axes/model.mlir b/iree_tests/onnx/node/generated/test_slice_negative_axes/model.mlir index 2aa603f55..ca75d0c2c 100644 --- a/iree_tests/onnx/node/generated/test_slice_negative_axes/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice_negative_axes/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice_negative_axes(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>, %arg3: !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice_negative_axes(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[3],si64>, %arg2: !torch.vtensor<[3],si64>, %arg3: !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2, %arg3) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[3],si64>) -> !torch.vtensor<[20,10,1],f32> return %0 : !torch.vtensor<[20,10,1],f32> diff --git a/iree_tests/onnx/node/generated/test_slice_start_out_of_bounds/model.mlir b/iree_tests/onnx/node/generated/test_slice_start_out_of_bounds/model.mlir index a0c1c1c2c..82bf83e99 100644 --- a/iree_tests/onnx/node/generated/test_slice_start_out_of_bounds/model.mlir +++ b/iree_tests/onnx/node/generated/test_slice_start_out_of_bounds/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_slice_start_out_of_bounds(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[1],si64>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],si64>, %arg4: !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,0,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_slice_start_out_of_bounds(%arg0: !torch.vtensor<[20,10,5],f32>, %arg1: !torch.vtensor<[1],si64>, %arg2: !torch.vtensor<[1],si64>, %arg3: !torch.vtensor<[1],si64>, %arg4: !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,0,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Slice"(%arg0, %arg1, %arg2, %arg3, %arg4) : (!torch.vtensor<[20,10,5],f32>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>, !torch.vtensor<[1],si64>) -> !torch.vtensor<[20,0,5],f32> return %0 : !torch.vtensor<[20,0,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_axis_0/model.mlir b/iree_tests/onnx/node/generated/test_softmax_axis_0/model.mlir index dc7033b93..01eb6404f 100644 --- a/iree_tests/onnx/node/generated/test_softmax_axis_0/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_axis_0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_axis_0(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_axis_0(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softmax"(%arg0) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_axis_0_expanded/model.mlir b/iree_tests/onnx/node/generated/test_softmax_axis_0_expanded/model.mlir index c5eff7c3e..bf9080181 100644 --- a/iree_tests/onnx/node/generated/test_softmax_axis_0_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_axis_0_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_axis_0_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_axis_0_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [0 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[1,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_axis_1/model.mlir b/iree_tests/onnx/node/generated/test_softmax_axis_1/model.mlir index d0aa78d2b..714f3e686 100644 --- a/iree_tests/onnx/node/generated/test_softmax_axis_1/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_axis_1/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_axis_1(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_axis_1(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softmax"(%arg0) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_axis_1_expanded/model.mlir b/iree_tests/onnx/node/generated/test_softmax_axis_1_expanded/model.mlir index de1d880c5..ea23eb75c 100644 --- a/iree_tests/onnx/node/generated/test_softmax_axis_1_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_axis_1_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_axis_1_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_axis_1_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,1,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_axis_2/model.mlir b/iree_tests/onnx/node/generated/test_softmax_axis_2/model.mlir index 37eff0de6..7d53ee44c 100644 --- a/iree_tests/onnx/node/generated/test_softmax_axis_2/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_axis_2/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_axis_2(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_axis_2(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softmax"(%arg0) {torch.onnx.axis = 2 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_axis_2_expanded/model.mlir b/iree_tests/onnx/node/generated/test_softmax_axis_2_expanded/model.mlir index 5c1fdfadb..9ee87e8b5 100644 --- a/iree_tests/onnx/node/generated/test_softmax_axis_2_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_axis_2_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_axis_2_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_axis_2_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<2> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [2 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,1],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_default_axis/model.mlir b/iree_tests/onnx/node/generated/test_softmax_default_axis/model.mlir index 8496502b8..60cb0ae2e 100644 --- a/iree_tests/onnx/node/generated/test_softmax_default_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_default_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_default_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_default_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softmax"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_default_axis_expanded/model.mlir b/iree_tests/onnx/node/generated/test_softmax_default_axis_expanded/model.mlir index 449666128..d629b84fb 100644 --- a/iree_tests/onnx/node/generated/test_softmax_default_axis_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_default_axis_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_default_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_default_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,1],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_example/model.mlir b/iree_tests/onnx/node/generated/test_softmax_example/model.mlir index 60454fe51..30524f5e4 100644 --- a/iree_tests/onnx/node/generated/test_softmax_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_example(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_example(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softmax"(%arg0) : (!torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> return %0 : !torch.vtensor<[1,3],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_example_expanded/model.mlir b/iree_tests/onnx/node/generated/test_softmax_example_expanded/model.mlir index 7a2a6ac9e..638a06071 100644 --- a/iree_tests/onnx/node/generated/test_softmax_example_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_example_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_example_expanded(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_example_expanded(%arg0: !torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[1,3],f32>) -> !torch.vtensor<[1,1],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_large_number/model.mlir b/iree_tests/onnx/node/generated/test_softmax_large_number/model.mlir index 8fa6d49ef..66f8f1da4 100644 --- a/iree_tests/onnx/node/generated/test_softmax_large_number/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_large_number/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_large_number(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_large_number(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softmax"(%arg0) : (!torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> return %0 : !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_large_number_expanded/model.mlir b/iree_tests/onnx/node/generated/test_softmax_large_number_expanded/model.mlir index 470203703..26ba39174 100644 --- a/iree_tests/onnx/node/generated/test_softmax_large_number_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_large_number_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_large_number_expanded(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_large_number_expanded(%arg0: !torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[2,4],f32>) -> !torch.vtensor<[2,1],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_softmax_negative_axis/model.mlir index 9e2143b65..565fc1b21 100644 --- a/iree_tests/onnx/node/generated/test_softmax_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_negative_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_negative_axis(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softmax"(%arg0) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softmax_negative_axis_expanded/model.mlir b/iree_tests/onnx/node/generated/test_softmax_negative_axis_expanded/model.mlir index 611c827b4..5c2f33803 100644 --- a/iree_tests/onnx/node/generated/test_softmax_negative_axis_expanded/model.mlir +++ b/iree_tests/onnx/node/generated/test_softmax_negative_axis_expanded/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softmax_negative_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softmax_negative_axis_expanded(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<-1> : tensor<1xsi64>} : () -> !torch.vtensor<[1],si64> %1 = torch.operator "onnx.ReduceMax"(%arg0) {torch.onnx.axes = [-1 : si64], torch.onnx.keepdims = 1 : si64} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,1],f32> diff --git a/iree_tests/onnx/node/generated/test_softplus/model.mlir b/iree_tests/onnx/node/generated/test_softplus/model.mlir index 7d4963ffc..68cfa3874 100644 --- a/iree_tests/onnx/node/generated/test_softplus/model.mlir +++ b/iree_tests/onnx/node/generated/test_softplus/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softplus(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softplus(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softplus"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softplus_example/model.mlir b/iree_tests/onnx/node/generated/test_softplus_example/model.mlir index 7883bbfe5..3effbb6bd 100644 --- a/iree_tests/onnx/node/generated/test_softplus_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_softplus_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softplus_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softplus_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softplus"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_softsign/model.mlir b/iree_tests/onnx/node/generated/test_softsign/model.mlir index efbf80ae7..1a64706fc 100644 --- a/iree_tests/onnx/node/generated/test_softsign/model.mlir +++ b/iree_tests/onnx/node/generated/test_softsign/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softsign(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softsign(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softsign"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_softsign_example/model.mlir b/iree_tests/onnx/node/generated/test_softsign_example/model.mlir index 289f9ef99..d813c2a54 100644 --- a/iree_tests/onnx/node/generated/test_softsign_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_softsign_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_softsign_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 1 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_softsign_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Softsign"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_spacetodepth/model.mlir b/iree_tests/onnx/node/generated/test_spacetodepth/model.mlir index a6390e33a..de87234f5 100644 --- a/iree_tests/onnx/node/generated/test_spacetodepth/model.mlir +++ b/iree_tests/onnx/node/generated/test_spacetodepth/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_spacetodepth(%arg0: !torch.vtensor<[2,2,6,6],f32>) -> !torch.vtensor<[2,8,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_spacetodepth(%arg0: !torch.vtensor<[2,2,6,6],f32>) -> !torch.vtensor<[2,8,3,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SpaceToDepth"(%arg0) {torch.onnx.blocksize = 2 : si64} : (!torch.vtensor<[2,2,6,6],f32>) -> !torch.vtensor<[2,8,3,3],f32> return %0 : !torch.vtensor<[2,8,3,3],f32> diff --git a/iree_tests/onnx/node/generated/test_spacetodepth_example/model.mlir b/iree_tests/onnx/node/generated/test_spacetodepth_example/model.mlir index de79681e0..d9cf353f5 100644 --- a/iree_tests/onnx/node/generated/test_spacetodepth_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_spacetodepth_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_spacetodepth_example(%arg0: !torch.vtensor<[1,1,4,6],f32>) -> !torch.vtensor<[1,4,2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_spacetodepth_example(%arg0: !torch.vtensor<[1,1,4,6],f32>) -> !torch.vtensor<[1,4,2,3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.SpaceToDepth"(%arg0) {torch.onnx.blocksize = 2 : si64} : (!torch.vtensor<[1,1,4,6],f32>) -> !torch.vtensor<[1,4,2,3],f32> return %0 : !torch.vtensor<[1,4,2,3],f32> diff --git a/iree_tests/onnx/node/generated/test_split_equal_parts_1d_opset13/model.mlir b/iree_tests/onnx/node/generated/test_split_equal_parts_1d_opset13/model.mlir index d00231a15..d4ace478a 100644 --- a/iree_tests/onnx/node/generated/test_split_equal_parts_1d_opset13/model.mlir +++ b/iree_tests/onnx/node/generated/test_split_equal_parts_1d_opset13/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_split_equal_parts_1d_opset13(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_split_equal_parts_1d_opset13(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.Split"(%arg0) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[6],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) return %0#0, %0#1, %0#2 : !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_split_equal_parts_2d_opset13/model.mlir b/iree_tests/onnx/node/generated/test_split_equal_parts_2d_opset13/model.mlir index c35d9a568..a7f20824a 100644 --- a/iree_tests/onnx/node/generated/test_split_equal_parts_2d_opset13/model.mlir +++ b/iree_tests/onnx/node/generated/test_split_equal_parts_2d_opset13/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_split_equal_parts_2d_opset13(%arg0: !torch.vtensor<[2,6],f32>) -> (!torch.vtensor<[2,3],f32>, !torch.vtensor<[2,3],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_split_equal_parts_2d_opset13(%arg0: !torch.vtensor<[2,6],f32>) -> (!torch.vtensor<[2,3],f32>, !torch.vtensor<[2,3],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Split"(%arg0) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[2,6],f32>) -> (!torch.vtensor<[2,3],f32>, !torch.vtensor<[2,3],f32>) return %0#0, %0#1 : !torch.vtensor<[2,3],f32>, !torch.vtensor<[2,3],f32> diff --git a/iree_tests/onnx/node/generated/test_split_equal_parts_default_axis_opset13/model.mlir b/iree_tests/onnx/node/generated/test_split_equal_parts_default_axis_opset13/model.mlir index 320bcea4e..1d991056a 100644 --- a/iree_tests/onnx/node/generated/test_split_equal_parts_default_axis_opset13/model.mlir +++ b/iree_tests/onnx/node/generated/test_split_equal_parts_default_axis_opset13/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_split_equal_parts_default_axis_opset13(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_split_equal_parts_default_axis_opset13(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.Split"(%arg0) : (!torch.vtensor<[6],f32>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>) return %0#0, %0#1, %0#2 : !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32>, !torch.vtensor<[2],f32> diff --git a/iree_tests/onnx/node/generated/test_split_variable_parts_1d_opset13/model.mlir b/iree_tests/onnx/node/generated/test_split_variable_parts_1d_opset13/model.mlir index f12efa2e4..0971378cf 100644 --- a/iree_tests/onnx/node/generated/test_split_variable_parts_1d_opset13/model.mlir +++ b/iree_tests/onnx/node/generated/test_split_variable_parts_1d_opset13/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_split_variable_parts_1d_opset13(%arg0: !torch.vtensor<[6],f32>, %arg1: !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_split_variable_parts_1d_opset13(%arg0: !torch.vtensor<[6],f32>, %arg1: !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Split"(%arg0, %arg1) {torch.onnx.axis = 0 : si64} : (!torch.vtensor<[6],f32>, !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[4],f32>) return %0#0, %0#1 : !torch.vtensor<[2],f32>, !torch.vtensor<[4],f32> diff --git a/iree_tests/onnx/node/generated/test_split_variable_parts_2d_opset13/model.mlir b/iree_tests/onnx/node/generated/test_split_variable_parts_2d_opset13/model.mlir index 42bf63a64..f7985de47 100644 --- a/iree_tests/onnx/node/generated/test_split_variable_parts_2d_opset13/model.mlir +++ b/iree_tests/onnx/node/generated/test_split_variable_parts_2d_opset13/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_split_variable_parts_2d_opset13(%arg0: !torch.vtensor<[2,6],f32>, %arg1: !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_split_variable_parts_2d_opset13(%arg0: !torch.vtensor<[2,6],f32>, %arg1: !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Split"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[2,6],f32>, !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2,4],f32>) return %0#0, %0#1 : !torch.vtensor<[2,2],f32>, !torch.vtensor<[2,4],f32> diff --git a/iree_tests/onnx/node/generated/test_split_variable_parts_default_axis_opset13/model.mlir b/iree_tests/onnx/node/generated/test_split_variable_parts_default_axis_opset13/model.mlir index 21625b643..5525da9e5 100644 --- a/iree_tests/onnx/node/generated/test_split_variable_parts_default_axis_opset13/model.mlir +++ b/iree_tests/onnx/node/generated/test_split_variable_parts_default_axis_opset13/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_split_variable_parts_default_axis_opset13(%arg0: !torch.vtensor<[6],f32>, %arg1: !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_split_variable_parts_default_axis_opset13(%arg0: !torch.vtensor<[6],f32>, %arg1: !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[4],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Split"(%arg0, %arg1) : (!torch.vtensor<[6],f32>, !torch.vtensor<[2],si64>) -> (!torch.vtensor<[2],f32>, !torch.vtensor<[4],f32>) return %0#0, %0#1 : !torch.vtensor<[2],f32>, !torch.vtensor<[4],f32> diff --git a/iree_tests/onnx/node/generated/test_split_zero_size_splits_opset13/model.mlir b/iree_tests/onnx/node/generated/test_split_zero_size_splits_opset13/model.mlir index fe4774abd..d087467b3 100644 --- a/iree_tests/onnx/node/generated/test_split_zero_size_splits_opset13/model.mlir +++ b/iree_tests/onnx/node/generated/test_split_zero_size_splits_opset13/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_split_zero_size_splits_opset13(%arg0: !torch.vtensor<[0],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[0],f32>, !torch.vtensor<[0],f32>, !torch.vtensor<[0],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_split_zero_size_splits_opset13(%arg0: !torch.vtensor<[0],f32>, %arg1: !torch.vtensor<[3],si64>) -> (!torch.vtensor<[0],f32>, !torch.vtensor<[0],f32>, !torch.vtensor<[0],f32>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:3 = torch.operator "onnx.Split"(%arg0, %arg1) : (!torch.vtensor<[0],f32>, !torch.vtensor<[3],si64>) -> (!torch.vtensor<[0],f32>, !torch.vtensor<[0],f32>, !torch.vtensor<[0],f32>) return %0#0, %0#1, %0#2 : !torch.vtensor<[0],f32>, !torch.vtensor<[0],f32>, !torch.vtensor<[0],f32> diff --git a/iree_tests/onnx/node/generated/test_sqrt/model.mlir b/iree_tests/onnx/node/generated/test_sqrt/model.mlir index 99b5f1a37..a530b389d 100644 --- a/iree_tests/onnx/node/generated/test_sqrt/model.mlir +++ b/iree_tests/onnx/node/generated/test_sqrt/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sqrt(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sqrt(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sqrt"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sqrt_example/model.mlir b/iree_tests/onnx/node/generated/test_sqrt_example/model.mlir index 0ae2b8b40..15445cec3 100644 --- a/iree_tests/onnx/node/generated/test_sqrt_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_sqrt_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sqrt_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sqrt_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sqrt"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_lower/model.mlir b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_lower/model.mlir index 08e7811f6..f742ad9cb 100644 --- a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_lower/model.mlir +++ b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_lower/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_strnormalizer_export_monday_casesensintive_lower(%arg0: !torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_strnormalizer_export_monday_casesensintive_lower(%arg0: !torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.StringNormalizer"(%arg0) {torch.onnx.case_change_action = "LOWER", torch.onnx.is_case_sensitive = 1 : si64, torch.onnx.stopwords = ["monday"]} : (!torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> return %0 : !torch.vtensor<[3],!torch.str> diff --git a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_nochangecase/model.mlir b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_nochangecase/model.mlir index 0e1054d7b..20ddb1a4e 100644 --- a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_nochangecase/model.mlir +++ b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_nochangecase/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_strnormalizer_export_monday_casesensintive_nochangecase(%arg0: !torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_strnormalizer_export_monday_casesensintive_nochangecase(%arg0: !torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.StringNormalizer"(%arg0) {torch.onnx.is_case_sensitive = 1 : si64, torch.onnx.stopwords = ["monday"]} : (!torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> return %0 : !torch.vtensor<[3],!torch.str> diff --git a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_upper/model.mlir b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_upper/model.mlir index 6b64fbe84..43cdbb9c6 100644 --- a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_upper/model.mlir +++ b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_casesensintive_upper/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_strnormalizer_export_monday_casesensintive_upper(%arg0: !torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_strnormalizer_export_monday_casesensintive_upper(%arg0: !torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.StringNormalizer"(%arg0) {torch.onnx.case_change_action = "UPPER", torch.onnx.is_case_sensitive = 1 : si64, torch.onnx.stopwords = ["monday"]} : (!torch.vtensor<[4],!torch.str>) -> !torch.vtensor<[3],!torch.str> return %0 : !torch.vtensor<[3],!torch.str> diff --git a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_empty_output/model.mlir b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_empty_output/model.mlir index 58537d583..e871cbb58 100644 --- a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_empty_output/model.mlir +++ b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_empty_output/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_strnormalizer_export_monday_empty_output(%arg0: !torch.vtensor<[2],!torch.str>) -> !torch.vtensor<[1],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_strnormalizer_export_monday_empty_output(%arg0: !torch.vtensor<[2],!torch.str>) -> !torch.vtensor<[1],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.StringNormalizer"(%arg0) {torch.onnx.case_change_action = "UPPER", torch.onnx.is_case_sensitive = 1 : si64, torch.onnx.stopwords = ["monday"]} : (!torch.vtensor<[2],!torch.str>) -> !torch.vtensor<[1],!torch.str> return %0 : !torch.vtensor<[1],!torch.str> diff --git a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_insensintive_upper_twodim/model.mlir b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_insensintive_upper_twodim/model.mlir index 4bb09e3bf..c93142941 100644 --- a/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_insensintive_upper_twodim/model.mlir +++ b/iree_tests/onnx/node/generated/test_strnormalizer_export_monday_insensintive_upper_twodim/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_strnormalizer_export_monday_insensintive_upper_twodim(%arg0: !torch.vtensor<[1,6],!torch.str>) -> !torch.vtensor<[1,4],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_strnormalizer_export_monday_insensintive_upper_twodim(%arg0: !torch.vtensor<[1,6],!torch.str>) -> !torch.vtensor<[1,4],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.StringNormalizer"(%arg0) {torch.onnx.case_change_action = "UPPER", torch.onnx.stopwords = ["monday"]} : (!torch.vtensor<[1,6],!torch.str>) -> !torch.vtensor<[1,4],!torch.str> return %0 : !torch.vtensor<[1,4],!torch.str> diff --git a/iree_tests/onnx/node/generated/test_strnormalizer_nostopwords_nochangecase/model.mlir b/iree_tests/onnx/node/generated/test_strnormalizer_nostopwords_nochangecase/model.mlir index f1cf9b262..4c7377bdf 100644 --- a/iree_tests/onnx/node/generated/test_strnormalizer_nostopwords_nochangecase/model.mlir +++ b/iree_tests/onnx/node/generated/test_strnormalizer_nostopwords_nochangecase/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_strnormalizer_nostopwords_nochangecase(%arg0: !torch.vtensor<[2],!torch.str>) -> !torch.vtensor<[2],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_strnormalizer_nostopwords_nochangecase(%arg0: !torch.vtensor<[2],!torch.str>) -> !torch.vtensor<[2],!torch.str> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.StringNormalizer"(%arg0) {torch.onnx.is_case_sensitive = 1 : si64} : (!torch.vtensor<[2],!torch.str>) -> !torch.vtensor<[2],!torch.str> return %0 : !torch.vtensor<[2],!torch.str> diff --git a/iree_tests/onnx/node/generated/test_sub/model.mlir b/iree_tests/onnx/node/generated/test_sub/model.mlir index ca70e9d76..798216ae4 100644 --- a/iree_tests/onnx/node/generated/test_sub/model.mlir +++ b/iree_tests/onnx/node/generated/test_sub/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sub(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sub(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sub"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sub_bcast/model.mlir b/iree_tests/onnx/node/generated/test_sub_bcast/model.mlir index 716c7de9b..940b7b2b5 100644 --- a/iree_tests/onnx/node/generated/test_sub_bcast/model.mlir +++ b/iree_tests/onnx/node/generated/test_sub_bcast/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sub_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sub_bcast(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sub"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_sub_example/model.mlir b/iree_tests/onnx/node/generated/test_sub_example/model.mlir index 5cee5de62..d40725a3a 100644 --- a/iree_tests/onnx/node/generated/test_sub_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_sub_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sub_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sub_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sub"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_sub_uint8/model.mlir b/iree_tests/onnx/node/generated/test_sub_uint8/model.mlir index 91fe48e0a..0f82c1d50 100644 --- a/iree_tests/onnx/node/generated/test_sub_uint8/model.mlir +++ b/iree_tests/onnx/node/generated/test_sub_uint8/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sub_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sub_uint8(%arg0: !torch.vtensor<[3,4,5],ui8>, %arg1: !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sub"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],ui8>, !torch.vtensor<[3,4,5],ui8>) -> !torch.vtensor<[3,4,5],ui8> return %0 : !torch.vtensor<[3,4,5],ui8> diff --git a/iree_tests/onnx/node/generated/test_sum_example/model.mlir b/iree_tests/onnx/node/generated/test_sum_example/model.mlir index 372696160..c8fe0ad2f 100644 --- a/iree_tests/onnx/node/generated/test_sum_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_sum_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sum_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sum_example(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>, %arg2: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sum"(%arg0, %arg1, %arg2) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_sum_one_input/model.mlir b/iree_tests/onnx/node/generated/test_sum_one_input/model.mlir index 59ba0e315..1a2936c23 100644 --- a/iree_tests/onnx/node/generated/test_sum_one_input/model.mlir +++ b/iree_tests/onnx/node/generated/test_sum_one_input/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sum_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sum_one_input(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sum"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_sum_two_inputs/model.mlir b/iree_tests/onnx/node/generated/test_sum_two_inputs/model.mlir index c2d2c1335..d6b02ae42 100644 --- a/iree_tests/onnx/node/generated/test_sum_two_inputs/model.mlir +++ b/iree_tests/onnx/node/generated/test_sum_two_inputs/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_sum_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_sum_two_inputs(%arg0: !torch.vtensor<[3],f32>, %arg1: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Sum"(%arg0, %arg1) : (!torch.vtensor<[3],f32>, !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_tan/model.mlir b/iree_tests/onnx/node/generated/test_tan/model.mlir index 253e50ba3..09284f9de 100644 --- a/iree_tests/onnx/node/generated/test_tan/model.mlir +++ b/iree_tests/onnx/node/generated/test_tan/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tan(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tan(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Tan"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_tan_example/model.mlir b/iree_tests/onnx/node/generated/test_tan_example/model.mlir index 330f19886..c5e50c59a 100644 --- a/iree_tests/onnx/node/generated/test_tan_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_tan_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tan_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tan_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Tan"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_tanh/model.mlir b/iree_tests/onnx/node/generated/test_tanh/model.mlir index 2bd6b1309..6a71faf2e 100644 --- a/iree_tests/onnx/node/generated/test_tanh/model.mlir +++ b/iree_tests/onnx/node/generated/test_tanh/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tanh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tanh(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Tanh"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_tanh_example/model.mlir b/iree_tests/onnx/node/generated/test_tanh_example/model.mlir index 8e61d5e2b..af28a43d2 100644 --- a/iree_tests/onnx/node/generated/test_tanh_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_tanh_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tanh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tanh_example(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Tanh"(%arg0) : (!torch.vtensor<[3],f32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip0/model.mlir b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip0/model.mlir index 09c7f2560..642be88c2 100644 --- a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip0/model.mlir +++ b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tfidfvectorizer_tf_batch_onlybigrams_skip0(%arg0: !torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tfidfvectorizer_tf_batch_onlybigrams_skip0(%arg0: !torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.TfIdfVectorizer"(%arg0) {torch.onnx.max_gram_length = 2 : si64, torch.onnx.max_skip_count = 0 : si64, torch.onnx.min_gram_length = 2 : si64, torch.onnx.mode = "TF", torch.onnx.ngram_counts = [0 : si64, 4 : si64], torch.onnx.ngram_indexes = [0 : si64, 1 : si64, 2 : si64, 3 : si64, 4 : si64, 5 : si64, 6 : si64], torch.onnx.pool_int64s = [2 : si64, 3 : si64, 5 : si64, 4 : si64, 5 : si64, 6 : si64, 7 : si64, 8 : si64, 6 : si64, 7 : si64]} : (!torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> return %0 : !torch.vtensor<[2,7],f32> diff --git a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip5/model.mlir b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip5/model.mlir index 1c003c77b..c812274b9 100644 --- a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip5/model.mlir +++ b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_onlybigrams_skip5/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tfidfvectorizer_tf_batch_onlybigrams_skip5(%arg0: !torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tfidfvectorizer_tf_batch_onlybigrams_skip5(%arg0: !torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.TfIdfVectorizer"(%arg0) {torch.onnx.max_gram_length = 2 : si64, torch.onnx.max_skip_count = 5 : si64, torch.onnx.min_gram_length = 2 : si64, torch.onnx.mode = "TF", torch.onnx.ngram_counts = [0 : si64, 4 : si64], torch.onnx.ngram_indexes = [0 : si64, 1 : si64, 2 : si64, 3 : si64, 4 : si64, 5 : si64, 6 : si64], torch.onnx.pool_int64s = [2 : si64, 3 : si64, 5 : si64, 4 : si64, 5 : si64, 6 : si64, 7 : si64, 8 : si64, 6 : si64, 7 : si64]} : (!torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> return %0 : !torch.vtensor<[2,7],f32> diff --git a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_uniandbigrams_skip5/model.mlir b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_uniandbigrams_skip5/model.mlir index 60044e8d7..3bd3c53e5 100644 --- a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_uniandbigrams_skip5/model.mlir +++ b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_batch_uniandbigrams_skip5/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tfidfvectorizer_tf_batch_uniandbigrams_skip5(%arg0: !torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tfidfvectorizer_tf_batch_uniandbigrams_skip5(%arg0: !torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.TfIdfVectorizer"(%arg0) {torch.onnx.max_gram_length = 2 : si64, torch.onnx.max_skip_count = 5 : si64, torch.onnx.min_gram_length = 1 : si64, torch.onnx.mode = "TF", torch.onnx.ngram_counts = [0 : si64, 4 : si64], torch.onnx.ngram_indexes = [0 : si64, 1 : si64, 2 : si64, 3 : si64, 4 : si64, 5 : si64, 6 : si64], torch.onnx.pool_int64s = [2 : si64, 3 : si64, 5 : si64, 4 : si64, 5 : si64, 6 : si64, 7 : si64, 8 : si64, 6 : si64, 7 : si64]} : (!torch.vtensor<[2,6],si32>) -> !torch.vtensor<[2,7],f32> return %0 : !torch.vtensor<[2,7],f32> diff --git a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_only_bigrams_skip0/model.mlir b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_only_bigrams_skip0/model.mlir index 1a1a08510..aafe7fa1c 100644 --- a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_only_bigrams_skip0/model.mlir +++ b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_only_bigrams_skip0/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tfidfvectorizer_tf_only_bigrams_skip0(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tfidfvectorizer_tf_only_bigrams_skip0(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.TfIdfVectorizer"(%arg0) {torch.onnx.max_gram_length = 2 : si64, torch.onnx.max_skip_count = 0 : si64, torch.onnx.min_gram_length = 2 : si64, torch.onnx.mode = "TF", torch.onnx.ngram_counts = [0 : si64, 4 : si64], torch.onnx.ngram_indexes = [0 : si64, 1 : si64, 2 : si64, 3 : si64, 4 : si64, 5 : si64, 6 : si64], torch.onnx.pool_int64s = [2 : si64, 3 : si64, 5 : si64, 4 : si64, 5 : si64, 6 : si64, 7 : si64, 8 : si64, 6 : si64, 7 : si64]} : (!torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> return %0 : !torch.vtensor<[7],f32> diff --git a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_levelempty/model.mlir b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_levelempty/model.mlir index a10c57c32..85d9c13ea 100644 --- a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_levelempty/model.mlir +++ b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_levelempty/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tfidfvectorizer_tf_onlybigrams_levelempty(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tfidfvectorizer_tf_onlybigrams_levelempty(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[3],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.TfIdfVectorizer"(%arg0) {torch.onnx.max_gram_length = 2 : si64, torch.onnx.max_skip_count = 0 : si64, torch.onnx.min_gram_length = 2 : si64, torch.onnx.mode = "TF", torch.onnx.ngram_counts = [0 : si64, 0 : si64], torch.onnx.ngram_indexes = [0 : si64, 1 : si64, 2 : si64], torch.onnx.pool_int64s = [5 : si64, 6 : si64, 7 : si64, 8 : si64, 6 : si64, 7 : si64]} : (!torch.vtensor<[12],si32>) -> !torch.vtensor<[3],f32> return %0 : !torch.vtensor<[3],f32> diff --git a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_skip5/model.mlir b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_skip5/model.mlir index 1b2ccc760..221bf62ee 100644 --- a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_skip5/model.mlir +++ b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_onlybigrams_skip5/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tfidfvectorizer_tf_onlybigrams_skip5(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tfidfvectorizer_tf_onlybigrams_skip5(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.TfIdfVectorizer"(%arg0) {torch.onnx.max_gram_length = 2 : si64, torch.onnx.max_skip_count = 5 : si64, torch.onnx.min_gram_length = 2 : si64, torch.onnx.mode = "TF", torch.onnx.ngram_counts = [0 : si64, 4 : si64], torch.onnx.ngram_indexes = [0 : si64, 1 : si64, 2 : si64, 3 : si64, 4 : si64, 5 : si64, 6 : si64], torch.onnx.pool_int64s = [2 : si64, 3 : si64, 5 : si64, 4 : si64, 5 : si64, 6 : si64, 7 : si64, 8 : si64, 6 : si64, 7 : si64]} : (!torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> return %0 : !torch.vtensor<[7],f32> diff --git a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_uniandbigrams_skip5/model.mlir b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_uniandbigrams_skip5/model.mlir index ec1b4aa7f..890a7812a 100644 --- a/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_uniandbigrams_skip5/model.mlir +++ b/iree_tests/onnx/node/generated/test_tfidfvectorizer_tf_uniandbigrams_skip5/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tfidfvectorizer_tf_uniandbigrams_skip5(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tfidfvectorizer_tf_uniandbigrams_skip5(%arg0: !torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.TfIdfVectorizer"(%arg0) {torch.onnx.max_gram_length = 2 : si64, torch.onnx.max_skip_count = 5 : si64, torch.onnx.min_gram_length = 1 : si64, torch.onnx.mode = "TF", torch.onnx.ngram_counts = [0 : si64, 4 : si64], torch.onnx.ngram_indexes = [0 : si64, 1 : si64, 2 : si64, 3 : si64, 4 : si64, 5 : si64, 6 : si64], torch.onnx.pool_int64s = [2 : si64, 3 : si64, 5 : si64, 4 : si64, 5 : si64, 6 : si64, 7 : si64, 8 : si64, 6 : si64, 7 : si64]} : (!torch.vtensor<[12],si32>) -> !torch.vtensor<[7],f32> return %0 : !torch.vtensor<[7],f32> diff --git a/iree_tests/onnx/node/generated/test_thresholdedrelu/model.mlir b/iree_tests/onnx/node/generated/test_thresholdedrelu/model.mlir index 5dcd96c9b..8b8491ba0 100644 --- a/iree_tests/onnx/node/generated/test_thresholdedrelu/model.mlir +++ b/iree_tests/onnx/node/generated/test_thresholdedrelu/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_thresholdedrelu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_thresholdedrelu(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ThresholdedRelu"(%arg0) {torch.onnx.alpha = 2.000000e+00 : f32} : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_thresholdedrelu_default/model.mlir b/iree_tests/onnx/node/generated/test_thresholdedrelu_default/model.mlir index ded47ca9a..063c42efc 100644 --- a/iree_tests/onnx/node/generated/test_thresholdedrelu_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_thresholdedrelu_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_thresholdedrelu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_thresholdedrelu_default(%arg0: !torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ThresholdedRelu"(%arg0) : (!torch.vtensor<[3,4,5],f32>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_thresholdedrelu_example/model.mlir b/iree_tests/onnx/node/generated/test_thresholdedrelu_example/model.mlir index 64feea28f..d332167dc 100644 --- a/iree_tests/onnx/node/generated/test_thresholdedrelu_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_thresholdedrelu_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_thresholdedrelu_example(%arg0: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 10 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_thresholdedrelu_example(%arg0: !torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> attributes {torch.onnx_meta.ir_version = 5 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.ThresholdedRelu"(%arg0) {torch.onnx.alpha = 2.000000e+00 : f32} : (!torch.vtensor<[5],f32>) -> !torch.vtensor<[5],f32> return %0 : !torch.vtensor<[5],f32> diff --git a/iree_tests/onnx/node/generated/test_tile/model.mlir b/iree_tests/onnx/node/generated/test_tile/model.mlir index 9c12a322c..bdf1f1180 100644 --- a/iree_tests/onnx/node/generated/test_tile/model.mlir +++ b/iree_tests/onnx/node/generated/test_tile/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tile(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[4],si64>) -> !torch.vtensor<[14,18,16,10],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tile(%arg0: !torch.vtensor<[2,3,4,5],f32>, %arg1: !torch.vtensor<[4],si64>) -> !torch.vtensor<[14,18,16,10],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Tile"(%arg0, %arg1) : (!torch.vtensor<[2,3,4,5],f32>, !torch.vtensor<[4],si64>) -> !torch.vtensor<[14,18,16,10],f32> return %0 : !torch.vtensor<[14,18,16,10],f32> diff --git a/iree_tests/onnx/node/generated/test_tile_precomputed/model.mlir b/iree_tests/onnx/node/generated/test_tile_precomputed/model.mlir index b3914cf54..45fb7617d 100644 --- a/iree_tests/onnx/node/generated/test_tile_precomputed/model.mlir +++ b/iree_tests/onnx/node/generated/test_tile_precomputed/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tile_precomputed(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2],si64>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tile_precomputed(%arg0: !torch.vtensor<[2,2],f32>, %arg1: !torch.vtensor<[2],si64>) -> !torch.vtensor<[4,4],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Tile"(%arg0, %arg1) : (!torch.vtensor<[2,2],f32>, !torch.vtensor<[2],si64>) -> !torch.vtensor<[4,4],f32> return %0 : !torch.vtensor<[4,4],f32> diff --git a/iree_tests/onnx/node/generated/test_top_k/model.mlir b/iree_tests/onnx/node/generated/test_top_k/model.mlir index fe7cbb3b7..9a32c5108 100644 --- a/iree_tests/onnx/node/generated/test_top_k/model.mlir +++ b/iree_tests/onnx/node/generated/test_top_k/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_top_k(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_top_k(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.TopK"(%arg0, %arg1) {torch.onnx.axis = 1 : si64} : (!torch.vtensor<[3,4],f32>, !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) return %0#0, %0#1 : !torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_top_k_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_top_k_negative_axis/model.mlir index 1da4255e5..a2f14e74e 100644 --- a/iree_tests/onnx/node/generated/test_top_k_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_top_k_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_top_k_negative_axis(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_top_k_negative_axis(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.TopK"(%arg0, %arg1) {torch.onnx.axis = -1 : si64} : (!torch.vtensor<[3,4],f32>, !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) return %0#0, %0#1 : !torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_top_k_smallest/model.mlir b/iree_tests/onnx/node/generated/test_top_k_smallest/model.mlir index 76dba87f8..43720fa0b 100644 --- a/iree_tests/onnx/node/generated/test_top_k_smallest/model.mlir +++ b/iree_tests/onnx/node/generated/test_top_k_smallest/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_top_k_smallest(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_top_k_smallest(%arg0: !torch.vtensor<[3,4],f32>, %arg1: !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.TopK"(%arg0, %arg1) {torch.onnx.axis = 1 : si64, torch.onnx.largest = 0 : si64, torch.onnx.sorted = 1 : si64} : (!torch.vtensor<[3,4],f32>, !torch.vtensor<[1],si64>) -> (!torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64>) return %0#0, %0#1 : !torch.vtensor<[3,3],f32>, !torch.vtensor<[3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_training_dropout/model.mlir b/iree_tests/onnx/node/generated/test_training_dropout/model.mlir index 3527c89a2..c14deda5c 100644 --- a/iree_tests/onnx/node/generated/test_training_dropout/model.mlir +++ b/iree_tests/onnx/node/generated/test_training_dropout/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_training_dropout(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_training_dropout(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Dropout"(%arg0, %arg1, %arg2) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_training_dropout_default/model.mlir b/iree_tests/onnx/node/generated/test_training_dropout_default/model.mlir index e7db221a1..46228c808 100644 --- a/iree_tests/onnx/node/generated/test_training_dropout_default/model.mlir +++ b/iree_tests/onnx/node/generated/test_training_dropout_default/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_training_dropout_default(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_training_dropout_default(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Dropout"(%arg0, %arg1, %arg2) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_training_dropout_default_mask/model.mlir b/iree_tests/onnx/node/generated/test_training_dropout_default_mask/model.mlir index 322117700..4553f57c9 100644 --- a/iree_tests/onnx/node/generated/test_training_dropout_default_mask/model.mlir +++ b/iree_tests/onnx/node/generated/test_training_dropout_default_mask/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_training_dropout_default_mask(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_training_dropout_default_mask(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Dropout"(%arg0, %arg1, %arg2) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) return %0#0, %0#1 : !torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_training_dropout_mask/model.mlir b/iree_tests/onnx/node/generated/test_training_dropout_mask/model.mlir index 54c27ad9c..500799e34 100644 --- a/iree_tests/onnx/node/generated/test_training_dropout_mask/model.mlir +++ b/iree_tests/onnx/node/generated/test_training_dropout_mask/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_training_dropout_mask(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_training_dropout_mask(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Dropout"(%arg0, %arg1, %arg2) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) return %0#0, %0#1 : !torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio/model.mlir b/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio/model.mlir index c70b5461f..336033437 100644 --- a/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio/model.mlir +++ b/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_training_dropout_zero_ratio(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_training_dropout_zero_ratio(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Dropout"(%arg0, %arg1, %arg2) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],i1>) -> !torch.vtensor<[3,4,5],f32> return %0 : !torch.vtensor<[3,4,5],f32> diff --git a/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio_mask/model.mlir b/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio_mask/model.mlir index 0d2cbe0bf..1777bf5c9 100644 --- a/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio_mask/model.mlir +++ b/iree_tests/onnx/node/generated/test_training_dropout_zero_ratio_mask/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_training_dropout_zero_ratio_mask(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 13 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_training_dropout_zero_ratio_mask(%arg0: !torch.vtensor<[3,4,5],f32>, %arg1: !torch.vtensor<[],f32>, %arg2: !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:2 = torch.operator "onnx.Dropout"(%arg0, %arg1, %arg2) {torch.onnx.seed = 0 : si64} : (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[],f32>, !torch.vtensor<[],i1>) -> (!torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1>) return %0#0, %0#1 : !torch.vtensor<[3,4,5],f32>, !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_tril/model.mlir b/iree_tests/onnx/node/generated/test_tril/model.mlir index 7f0066622..fd52223d0 100644 --- a/iree_tests/onnx/node/generated/test_tril/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril(%arg0: !torch.vtensor<[4,5],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril(%arg0: !torch.vtensor<[4,5],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[4,5],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_neg/model.mlir b/iree_tests/onnx/node/generated/test_tril_neg/model.mlir index 8e2f5de5a..325fb7287 100644 --- a/iree_tests/onnx/node/generated/test_tril_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_neg(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_neg(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_one_row_neg/model.mlir b/iree_tests/onnx/node/generated/test_tril_one_row_neg/model.mlir index 647ee3981..c0190d55f 100644 --- a/iree_tests/onnx/node/generated/test_tril_one_row_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_one_row_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_one_row_neg(%arg0: !torch.vtensor<[3,1,5],si64>) -> !torch.vtensor<[3,1,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_one_row_neg(%arg0: !torch.vtensor<[3,1,5],si64>) -> !torch.vtensor<[3,1,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[3,1,5],si64>) -> !torch.vtensor<[3,1,5],si64> return %0 : !torch.vtensor<[3,1,5],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_out_neg/model.mlir b/iree_tests/onnx/node/generated/test_tril_out_neg/model.mlir index a646d9d03..4ad2f944e 100644 --- a/iree_tests/onnx/node/generated/test_tril_out_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_out_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_out_neg(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_out_neg(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_out_pos/model.mlir b/iree_tests/onnx/node/generated/test_tril_out_pos/model.mlir index 0cf8d68f6..afafca2ca 100644 --- a/iree_tests/onnx/node/generated/test_tril_out_pos/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_out_pos/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_out_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_out_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_pos/model.mlir b/iree_tests/onnx/node/generated/test_tril_pos/model.mlir index 4ccbe1c8d..ae4ab68ed 100644 --- a/iree_tests/onnx/node/generated/test_tril_pos/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_pos/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_square/model.mlir b/iree_tests/onnx/node/generated/test_tril_square/model.mlir index 73b014fa3..4da57c028 100644 --- a/iree_tests/onnx/node/generated/test_tril_square/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_square/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_square(%arg0: !torch.vtensor<[2,3,3],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_square(%arg0: !torch.vtensor<[2,3,3],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[2,3,3],si64>) -> !torch.vtensor<[2,3,3],si64> return %0 : !torch.vtensor<[2,3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_square_neg/model.mlir b/iree_tests/onnx/node/generated/test_tril_square_neg/model.mlir index 6588aedec..4fbf0fe5c 100644 --- a/iree_tests/onnx/node/generated/test_tril_square_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_square_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_square_neg(%arg0: !torch.vtensor<[2,3,3],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_square_neg(%arg0: !torch.vtensor<[2,3,3],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[2,3,3],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[2,3,3],si64> return %0 : !torch.vtensor<[2,3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_tril_zero/model.mlir b/iree_tests/onnx/node/generated/test_tril_zero/model.mlir index 11d576d6c..e9ff634fb 100644 --- a/iree_tests/onnx/node/generated/test_tril_zero/model.mlir +++ b/iree_tests/onnx/node/generated/test_tril_zero/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_tril_zero(%arg0: !torch.vtensor<[3,0,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[3,0,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_tril_zero(%arg0: !torch.vtensor<[3,0,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[3,0,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) {torch.onnx.upper = 0 : si64} : (!torch.vtensor<[3,0,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[3,0,5],si64> return %0 : !torch.vtensor<[3,0,5],si64> diff --git a/iree_tests/onnx/node/generated/test_triu/model.mlir b/iree_tests/onnx/node/generated/test_triu/model.mlir index 423626a65..581d0c17d 100644 --- a/iree_tests/onnx/node/generated/test_triu/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu(%arg0: !torch.vtensor<[4,5],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu(%arg0: !torch.vtensor<[4,5],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0) : (!torch.vtensor<[4,5],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_neg/model.mlir b/iree_tests/onnx/node/generated/test_triu_neg/model.mlir index e46c0728b..5abf6edc4 100644 --- a/iree_tests/onnx/node/generated/test_triu_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_neg(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_neg(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_one_row/model.mlir b/iree_tests/onnx/node/generated/test_triu_one_row/model.mlir index 0c8153526..17a2d4ae2 100644 --- a/iree_tests/onnx/node/generated/test_triu_one_row/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_one_row/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_one_row(%arg0: !torch.vtensor<[3,1,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[3,1,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_one_row(%arg0: !torch.vtensor<[3,1,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[3,1,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) : (!torch.vtensor<[3,1,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[3,1,5],si64> return %0 : !torch.vtensor<[3,1,5],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_out_neg_out/model.mlir b/iree_tests/onnx/node/generated/test_triu_out_neg_out/model.mlir index 887258d72..1a5c4737b 100644 --- a/iree_tests/onnx/node/generated/test_triu_out_neg_out/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_out_neg_out/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_out_neg_out(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_out_neg_out(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_out_pos/model.mlir b/iree_tests/onnx/node/generated/test_triu_out_pos/model.mlir index 0972be62c..dfa9126fe 100644 --- a/iree_tests/onnx/node/generated/test_triu_out_pos/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_out_pos/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_out_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_out_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_pos/model.mlir b/iree_tests/onnx/node/generated/test_triu_pos/model.mlir index d208eb4aa..97239bd34 100644 --- a/iree_tests/onnx/node/generated/test_triu_pos/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_pos/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_pos(%arg0: !torch.vtensor<[4,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) : (!torch.vtensor<[4,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[4,5],si64> return %0 : !torch.vtensor<[4,5],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_square/model.mlir b/iree_tests/onnx/node/generated/test_triu_square/model.mlir index f17b75094..c58780a4d 100644 --- a/iree_tests/onnx/node/generated/test_triu_square/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_square/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_square(%arg0: !torch.vtensor<[2,3,3],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_square(%arg0: !torch.vtensor<[2,3,3],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0) : (!torch.vtensor<[2,3,3],si64>) -> !torch.vtensor<[2,3,3],si64> return %0 : !torch.vtensor<[2,3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_square_neg/model.mlir b/iree_tests/onnx/node/generated/test_triu_square_neg/model.mlir index 955a70393..a74d9c32a 100644 --- a/iree_tests/onnx/node/generated/test_triu_square_neg/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_square_neg/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_square_neg(%arg0: !torch.vtensor<[2,3,3],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_square_neg(%arg0: !torch.vtensor<[2,3,3],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[2,3,3],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) : (!torch.vtensor<[2,3,3],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[2,3,3],si64> return %0 : !torch.vtensor<[2,3,3],si64> diff --git a/iree_tests/onnx/node/generated/test_triu_zero/model.mlir b/iree_tests/onnx/node/generated/test_triu_zero/model.mlir index 839ef5c1f..df4e38ccf 100644 --- a/iree_tests/onnx/node/generated/test_triu_zero/model.mlir +++ b/iree_tests/onnx/node/generated/test_triu_zero/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_triu_zero(%arg0: !torch.vtensor<[0,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[0,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 14 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_triu_zero(%arg0: !torch.vtensor<[0,5],si64>, %arg1: !torch.vtensor<[],si64>) -> !torch.vtensor<[0,5],si64> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Trilu"(%arg0, %arg1) : (!torch.vtensor<[0,5],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[0,5],si64> return %0 : !torch.vtensor<[0,5],si64> diff --git a/iree_tests/onnx/node/generated/test_unique_not_sorted_without_axis/model.mlir b/iree_tests/onnx/node/generated/test_unique_not_sorted_without_axis/model.mlir index 74fa0b759..55faa6a5f 100644 --- a/iree_tests/onnx/node/generated/test_unique_not_sorted_without_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_unique_not_sorted_without_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_unique_not_sorted_without_axis(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_unique_not_sorted_without_axis(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:4 = torch.operator "onnx.Unique"(%arg0) {torch.onnx.sorted = 0 : si64} : (!torch.vtensor<[6],f32>) -> (!torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64>) return %0#0, %0#1, %0#2, %0#3 : !torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64> diff --git a/iree_tests/onnx/node/generated/test_unique_sorted_with_axis/model.mlir b/iree_tests/onnx/node/generated/test_unique_sorted_with_axis/model.mlir index 5cf2d0620..321bf810e 100644 --- a/iree_tests/onnx/node/generated/test_unique_sorted_with_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_unique_sorted_with_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_unique_sorted_with_axis(%arg0: !torch.vtensor<[3,3],f32>) -> (!torch.vtensor<[2,3],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_unique_sorted_with_axis(%arg0: !torch.vtensor<[3,3],f32>) -> (!torch.vtensor<[2,3],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:4 = torch.operator "onnx.Unique"(%arg0) {torch.onnx.axis = 0 : si64, torch.onnx.sorted = 1 : si64} : (!torch.vtensor<[3,3],f32>) -> (!torch.vtensor<[2,3],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64>) return %0#0, %0#1, %0#2, %0#3 : !torch.vtensor<[2,3],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64> diff --git a/iree_tests/onnx/node/generated/test_unique_sorted_with_axis_3d/model.mlir b/iree_tests/onnx/node/generated/test_unique_sorted_with_axis_3d/model.mlir index b6c38456a..60ce22af3 100644 --- a/iree_tests/onnx/node/generated/test_unique_sorted_with_axis_3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_unique_sorted_with_axis_3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_unique_sorted_with_axis_3d(%arg0: !torch.vtensor<[2,4,2],f32>) -> (!torch.vtensor<[2,3,2],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[4],si64>, !torch.vtensor<[3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_unique_sorted_with_axis_3d(%arg0: !torch.vtensor<[2,4,2],f32>) -> (!torch.vtensor<[2,3,2],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[4],si64>, !torch.vtensor<[3],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:4 = torch.operator "onnx.Unique"(%arg0) {torch.onnx.axis = 1 : si64, torch.onnx.sorted = 1 : si64} : (!torch.vtensor<[2,4,2],f32>) -> (!torch.vtensor<[2,3,2],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[4],si64>, !torch.vtensor<[3],si64>) return %0#0, %0#1, %0#2, %0#3 : !torch.vtensor<[2,3,2],f32>, !torch.vtensor<[3],si64>, !torch.vtensor<[4],si64>, !torch.vtensor<[3],si64> diff --git a/iree_tests/onnx/node/generated/test_unique_sorted_with_negative_axis/model.mlir b/iree_tests/onnx/node/generated/test_unique_sorted_with_negative_axis/model.mlir index 1537829a5..e7d4cdfdf 100644 --- a/iree_tests/onnx/node/generated/test_unique_sorted_with_negative_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_unique_sorted_with_negative_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_unique_sorted_with_negative_axis(%arg0: !torch.vtensor<[3,3],f32>) -> (!torch.vtensor<[3,2],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_unique_sorted_with_negative_axis(%arg0: !torch.vtensor<[3,3],f32>) -> (!torch.vtensor<[3,2],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:4 = torch.operator "onnx.Unique"(%arg0) {torch.onnx.axis = -1 : si64, torch.onnx.sorted = 1 : si64} : (!torch.vtensor<[3,3],f32>) -> (!torch.vtensor<[3,2],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64>) return %0#0, %0#1, %0#2, %0#3 : !torch.vtensor<[3,2],f32>, !torch.vtensor<[2],si64>, !torch.vtensor<[3],si64>, !torch.vtensor<[2],si64> diff --git a/iree_tests/onnx/node/generated/test_unique_sorted_without_axis/model.mlir b/iree_tests/onnx/node/generated/test_unique_sorted_without_axis/model.mlir index eb23257eb..7796135f1 100644 --- a/iree_tests/onnx/node/generated/test_unique_sorted_without_axis/model.mlir +++ b/iree_tests/onnx/node/generated/test_unique_sorted_without_axis/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_unique_sorted_without_axis(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 11 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_unique_sorted_without_axis(%arg0: !torch.vtensor<[6],f32>) -> (!torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64>) attributes {torch.onnx_meta.ir_version = 6 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0:4 = torch.operator "onnx.Unique"(%arg0) : (!torch.vtensor<[6],f32>) -> (!torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64>) return %0#0, %0#1, %0#2, %0#3 : !torch.vtensor<[4],f32>, !torch.vtensor<[4],si64>, !torch.vtensor<[6],si64>, !torch.vtensor<[4],si64> diff --git a/iree_tests/onnx/node/generated/test_upsample_nearest/model.mlir b/iree_tests/onnx/node/generated/test_upsample_nearest/model.mlir index 28d40dff1..cbce08a92 100644 --- a/iree_tests/onnx/node/generated/test_upsample_nearest/model.mlir +++ b/iree_tests/onnx/node/generated/test_upsample_nearest/model.mlir @@ -1,8 +1,9 @@ module { - func.func @test_upsample_nearest(%arg0: !torch.vtensor<[1,1,2,2],f32>, %arg1: !torch.vtensor<[4],f32>) -> !torch.vtensor<[1,1,4,6],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 9 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_upsample_nearest(%arg0: !torch.vtensor<[1,1,2,2],f32>, %arg1: !torch.vtensor<[4],f32>) -> !torch.vtensor<[1,1,4,6],f32> attributes {torch.onnx_meta.ir_version = 4 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none - %0 = torch.operator "onnx.Upsample"(%arg0, %arg1) {torch.onnx.mode = "nearest"} : (!torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[4],f32>) -> !torch.vtensor<[1,1,4,6],f32> - return %0 : !torch.vtensor<[1,1,4,6],f32> + %0 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<[0.000000e+00, 0.000000e+00, 0.000000e+00, 0.000000e+00, 1.000000e+00, 1.000000e+00, 1.000000e+00, 1.000000e+00]> : tensor<8xf32>} : () -> !torch.vtensor<[8],f32> + %1 = torch.operator "onnx.Resize"(%arg0, %0, %arg1) {torch.onnx.mode = "nearest"} : (!torch.vtensor<[1,1,2,2],f32>, !torch.vtensor<[8],f32>, !torch.vtensor<[4],f32>) -> !torch.vtensor<[1,1,4,6],f32> + return %1 : !torch.vtensor<[1,1,4,6],f32> } } diff --git a/iree_tests/onnx/node/generated/test_where_example/model.mlir b/iree_tests/onnx/node/generated/test_where_example/model.mlir index 3f792dea0..94ceeed78 100644 --- a/iree_tests/onnx/node/generated/test_where_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_where_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_where_example(%arg0: !torch.vtensor<[2,2],i1>, %arg1: !torch.vtensor<[2,2],f32>, %arg2: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,2],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_where_example(%arg0: !torch.vtensor<[2,2],i1>, %arg1: !torch.vtensor<[2,2],f32>, %arg2: !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,2],f32> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Where"(%arg0, %arg1, %arg2) : (!torch.vtensor<[2,2],i1>, !torch.vtensor<[2,2],f32>, !torch.vtensor<[2,2],f32>) -> !torch.vtensor<[2,2],f32> return %0 : !torch.vtensor<[2,2],f32> diff --git a/iree_tests/onnx/node/generated/test_where_long_example/model.mlir b/iree_tests/onnx/node/generated/test_where_long_example/model.mlir index 7f34c464c..fa5e33a48 100644 --- a/iree_tests/onnx/node/generated/test_where_long_example/model.mlir +++ b/iree_tests/onnx/node/generated/test_where_long_example/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_where_long_example(%arg0: !torch.vtensor<[2,2],i1>, %arg1: !torch.vtensor<[2,2],si64>, %arg2: !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2,2],si64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 16 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_where_long_example(%arg0: !torch.vtensor<[2,2],i1>, %arg1: !torch.vtensor<[2,2],si64>, %arg2: !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2,2],si64> attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Where"(%arg0, %arg1, %arg2) : (!torch.vtensor<[2,2],i1>, !torch.vtensor<[2,2],si64>, !torch.vtensor<[2,2],si64>) -> !torch.vtensor<[2,2],si64> return %0 : !torch.vtensor<[2,2],si64> diff --git a/iree_tests/onnx/node/generated/test_xor2d/model.mlir b/iree_tests/onnx/node/generated/test_xor2d/model.mlir index 5740a1b0e..7df74880a 100644 --- a/iree_tests/onnx/node/generated/test_xor2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor2d(%arg0: !torch.vtensor<[3,4],i1>, %arg1: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor2d(%arg0: !torch.vtensor<[3,4],i1>, %arg1: !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[3,4],i1>, !torch.vtensor<[3,4],i1>) -> !torch.vtensor<[3,4],i1> return %0 : !torch.vtensor<[3,4],i1> diff --git a/iree_tests/onnx/node/generated/test_xor3d/model.mlir b/iree_tests/onnx/node/generated/test_xor3d/model.mlir index 676fc0ce7..f632e3959 100644 --- a/iree_tests/onnx/node/generated/test_xor3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor3d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor3d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[3,4,5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_xor4d/model.mlir b/iree_tests/onnx/node/generated/test_xor4d/model.mlir index a7cd50431..1273493cc 100644 --- a/iree_tests/onnx/node/generated/test_xor4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor4d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor4d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[3,4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_xor_bcast3v1d/model.mlir b/iree_tests/onnx/node/generated/test_xor_bcast3v1d/model.mlir index 0d094053f..6a53396e0 100644 --- a/iree_tests/onnx/node/generated/test_xor_bcast3v1d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor_bcast3v1d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor_bcast3v1d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor_bcast3v1d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_xor_bcast3v2d/model.mlir b/iree_tests/onnx/node/generated/test_xor_bcast3v2d/model.mlir index f13b1a89e..dd1a5b6a5 100644 --- a/iree_tests/onnx/node/generated/test_xor_bcast3v2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor_bcast3v2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor_bcast3v2d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor_bcast3v2d(%arg0: !torch.vtensor<[3,4,5],i1>, %arg1: !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[3,4,5],i1>, !torch.vtensor<[4,5],i1>) -> !torch.vtensor<[3,4,5],i1> return %0 : !torch.vtensor<[3,4,5],i1> diff --git a/iree_tests/onnx/node/generated/test_xor_bcast4v2d/model.mlir b/iree_tests/onnx/node/generated/test_xor_bcast4v2d/model.mlir index 7299ea2c0..fcbb7aabb 100644 --- a/iree_tests/onnx/node/generated/test_xor_bcast4v2d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor_bcast4v2d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor_bcast4v2d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor_bcast4v2d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_xor_bcast4v3d/model.mlir b/iree_tests/onnx/node/generated/test_xor_bcast4v3d/model.mlir index 3f6f48445..5ede0ae2e 100644 --- a/iree_tests/onnx/node/generated/test_xor_bcast4v3d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor_bcast4v3d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor_bcast4v3d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor_bcast4v3d(%arg0: !torch.vtensor<[3,4,5,6],i1>, %arg1: !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[3,4,5,6],i1>, !torch.vtensor<[4,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1> diff --git a/iree_tests/onnx/node/generated/test_xor_bcast4v4d/model.mlir b/iree_tests/onnx/node/generated/test_xor_bcast4v4d/model.mlir index 9383ea2d2..fd1caa0a3 100644 --- a/iree_tests/onnx/node/generated/test_xor_bcast4v4d/model.mlir +++ b/iree_tests/onnx/node/generated/test_xor_bcast4v4d/model.mlir @@ -1,5 +1,5 @@ module { - func.func @test_xor_bcast4v4d(%arg0: !torch.vtensor<[1,4,1,6],i1>, %arg1: !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 7 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { + func.func @test_xor_bcast4v4d(%arg0: !torch.vtensor<[1,4,1,6],i1>, %arg1: !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> attributes {torch.onnx_meta.ir_version = 3 : si64, torch.onnx_meta.opset_version = 17 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} { %none = torch.constant.none %0 = torch.operator "onnx.Xor"(%arg0, %arg1) : (!torch.vtensor<[1,4,1,6],i1>, !torch.vtensor<[3,1,5,6],i1>) -> !torch.vtensor<[3,4,5,6],i1> return %0 : !torch.vtensor<[3,4,5,6],i1>