xref: /aosp_15_r20/external/pytorch/test/onnx/expect/TestOperators.test_baddbmm.expect (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1ir_version: 8
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
5  node {
6    input: "onnx::MatMul_1"
7    input: "onnx::MatMul_2"
8    output: "onnx::Mul_4"
9    name: "MatMul_2"
10    op_type: "MatMul"
11  }
12  node {
13    output: "onnx::Mul_10"
14    name: "Constant_3"
15    op_type: "Constant"
16    attribute {
17      name: "value"
18      t {
19        data_type: 1
20        raw_data: "\000\000\200?"
21      }
22      type: TENSOR
23    }
24  }
25  node {
26    input: "onnx::Mul_4"
27    input: "onnx::Mul_10"
28    output: "onnx::Add_6"
29    name: "Mul_4"
30    op_type: "Mul"
31  }
32  node {
33    output: "onnx::Mul_11"
34    name: "Constant_5"
35    op_type: "Constant"
36    attribute {
37      name: "value"
38      t {
39        data_type: 1
40        raw_data: "\000\000\200?"
41      }
42      type: TENSOR
43    }
44  }
45  node {
46    input: "onnx::Mul_0"
47    input: "onnx::Mul_11"
48    output: "onnx::Add_8"
49    name: "Mul_6"
50    op_type: "Mul"
51  }
52  node {
53    input: "onnx::Add_6"
54    input: "onnx::Add_8"
55    output: "9"
56    name: "Add_7"
57    op_type: "Add"
58  }
59  name: "main_graph"
60  input {
61    name: "onnx::Mul_0"
62    type {
63      tensor_type {
64        elem_type: 1
65        shape {
66          dim {
67            dim_value: 10
68          }
69          dim {
70            dim_value: 3
71          }
72          dim {
73            dim_value: 5
74          }
75        }
76      }
77    }
78  }
79  input {
80    name: "onnx::MatMul_1"
81    type {
82      tensor_type {
83        elem_type: 1
84        shape {
85          dim {
86            dim_value: 10
87          }
88          dim {
89            dim_value: 3
90          }
91          dim {
92            dim_value: 4
93          }
94        }
95      }
96    }
97  }
98  input {
99    name: "onnx::MatMul_2"
100    type {
101      tensor_type {
102        elem_type: 1
103        shape {
104          dim {
105            dim_value: 10
106          }
107          dim {
108            dim_value: 4
109          }
110          dim {
111            dim_value: 5
112          }
113        }
114      }
115    }
116  }
117  output {
118    name: "9"
119    type {
120      tensor_type {
121        elem_type: 1
122        shape {
123          dim {
124            dim_value: 10
125          }
126          dim {
127            dim_value: 3
128          }
129          dim {
130            dim_value: 5
131          }
132        }
133      }
134    }
135  }
136}
137opset_import {
138  version: 17
139}
140