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