xref: /aosp_15_r20/external/pytorch/test/onnx/expect/TestOperators.test_softmaxcrossentropy.expect (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1ir_version: 7
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
5  node {
6    input: "input"
7    input: "target"
8    output: "2"
9    name: "SoftmaxCrossEntropyLoss_0"
10    op_type: "SoftmaxCrossEntropyLoss"
11    attribute {
12      name: "ignore_index"
13      i: -100
14      type: INT
15    }
16    attribute {
17      name: "reduction"
18      s: "mean"
19      type: STRING
20    }
21  }
22  name: "main_graph"
23  input {
24    name: "input"
25    type {
26      tensor_type {
27        elem_type: 1
28        shape {
29          dim {
30            dim_value: 3
31          }
32          dim {
33            dim_value: 5
34          }
35        }
36      }
37    }
38  }
39  input {
40    name: "target"
41    type {
42      tensor_type {
43        elem_type: 7
44        shape {
45          dim {
46            dim_value: 3
47          }
48        }
49      }
50    }
51  }
52  output {
53    name: "2"
54    type {
55      tensor_type {
56        elem_type: 1
57        shape {
58        }
59      }
60    }
61  }
62}
63opset_import {
64  version: 12
65}
66