• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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: "none"
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          dim {
36            dim_value: 2
37          }
38        }
39      }
40    }
41  }
42  input {
43    name: "target"
44    type {
45      tensor_type {
46        elem_type: 7
47        shape {
48          dim {
49            dim_value: 3
50          }
51          dim {
52            dim_value: 2
53          }
54        }
55      }
56    }
57  }
58  output {
59    name: "2"
60    type {
61      tensor_type {
62        elem_type: 1
63        shape {
64          dim {
65            dim_value: 3
66          }
67          dim {
68            dim_value: 2
69          }
70        }
71      }
72    }
73  }
74}
75opset_import {
76  version: 12
77}
78