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
    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
}
70
opset_import {
71
  version: 12
72
}
73