• 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    input: "weight"
9    output: "3"
10    name: "SoftmaxCrossEntropyLoss_0"
11    op_type: "SoftmaxCrossEntropyLoss"
12    attribute {
13      name: "ignore_index"
14      i: -100
15      type: INT
16    }
17    attribute {
18      name: "reduction"
19      s: "mean"
20      type: STRING
21    }
22  }
23  name: "main_graph"
24  initializer {
25    dims: 5
26    data_type: 1
27    name: "weight"
28    raw_data: "\334\204b?x\017\034\277C\300T?\246\205\346\275\227W\315\275"
29  }
30  input {
31    name: "input"
32    type {
33      tensor_type {
34        elem_type: 1
35        shape {
36          dim {
37            dim_value: 3
38          }
39          dim {
40            dim_value: 5
41          }
42        }
43      }
44    }
45  }
46  input {
47    name: "target"
48    type {
49      tensor_type {
50        elem_type: 7
51        shape {
52          dim {
53            dim_value: 3
54          }
55        }
56      }
57    }
58  }
59  output {
60    name: "3"
61    type {
62      tensor_type {
63        elem_type: 1
64        shape {
65        }
66      }
67    }
68  }
69}
70opset_import {
71  version: 12
72}
73