# The Epilogue, or endmost part of inception v3 comprised of Squeeze,
# Reshape, Shape and Softmax
#
# Only difference from original is input shape:
#   - original has unknown batch and 1001 channels [?, 1, 1, 1001]
#   - this test has 2 batches and 3 channels [2, 1, 1, 3]

node {
  name: "placeholder"
  op: "Placeholder"
  attr {
    key: "dtype"
    value { type: DT_FLOAT }
  }
  attr {
    key: "shape"
    value {
      shape {
        dim { size: 2 }
        dim { size: 1 }
        dim { size: 1 }
        dim { size: 3 }
      }
    }
  }
}
node {
  name: "squeeze"
  op: "Squeeze"
  input: "placeholder"
  attr {
    key: "T"
    value { type: DT_FLOAT }
  }
  attr {
    key: "squeeze_dims"
    value {
      list { i: 1 i: 2 }
    }
  }
}
node {
  name: "Reshape/shape"
  op: "Const"
  attr {
    key: "dtype"
    value { type: DT_INT32 }
  }
  attr {
    key: "value"
    value {
      tensor {
        dtype: DT_INT32
        tensor_shape {
          dim { size: 2 }
        }
        int_val: -1
        int_val: 3
      }
    }
  }
}
node {
  name: "reshape_1"
  op: "Reshape"
  input: "squeeze"
  input: "Reshape/shape"
  attr {
    key: "T"
    value { type: DT_FLOAT }
  }
  attr {
    key: "Tshape"
    value { type: DT_INT32 }
  }
}
node {
  name: "softmax"
  op: "Softmax"
  input: "reshape_1"
  attr {
    key: "T"
    value { type: DT_FLOAT }
  }
}
node {
  name: "shape"
  op: "Shape"
  input: "squeeze"
  attr {
    key: "T"
    value { type: DT_FLOAT }
  }
  attr {
    key: "out_type"
    value { type: DT_INT32 }
  }
}
node {
  name: "reshape_2"
  op: "Reshape"
  input: "softmax"
  input: "shape"
  attr {
    key: "T"
    value { type: DT_FLOAT }
  }
  attr {
    key: "Tshape"
    value { type: DT_INT32 }
  }
}