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
          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
        }
65
      }
66
    }
67
  }
68
}
69
opset_import {
70
  version: 12
71
}
72