batch_norm.pbtxt 1.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
type: "batch_norm"
def {
  inputs {
    name: "X"
  }
  inputs {
    name: "Scale"
  }
  inputs {
    name: "Bias"
  }
  inputs {
    name: "Mean"
  }
  inputs {
    name: "Variance"
  }
  outputs {
    name: "Y"
  }
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
  outputs {
    name: "MeanOut"
  }
  outputs {
    name: "VarianceOut"
  }
  outputs {
    name: "SavedMean"
  }
  outputs {
    name: "SavedVariance"
  }
  outputs {
    name: "ReserveSpace"
  }
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
  attrs {
    name: "epsilon"
    type: FLOAT
  }
}
extra {
  inputs {
    name: "MomentumTensor"
  }
  attrs {
    name: "is_test"
    type: BOOLEAN
  }
  attrs {
    name: "momentum"
    type: FLOAT
  }
  attrs {
    name: "data_layout"
    type: STRING
  }
  attrs {
    name: "use_mkldnn"
    type: BOOLEAN
  }
  attrs {
    name: "fuse_with_relu"
    type: BOOLEAN
  }
  attrs {
    name: "use_global_stats"
    type: BOOLEAN
  }
  attrs {
    name: "trainable_statistics"
    type: BOOLEAN
  }
  attrs {
    name: "op_role"
    type: INT
  }
  attrs {
    name: "op_role_var"
    type: STRINGS
  }
  attrs {
    name: "op_namescope"
    type: STRING
  }
  attrs {
    name: "op_callstack"
    type: STRINGS
  }
  attrs {
    name: "op_device"
    type: STRING
  }
}