pytorch

1
ir_version: 7
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
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
}
63
opset_import {
64
  version: 12
65
}
66