Ort infernce bug

[horheynm]

Fix for shape computation.

Before:
dim_param: "batch" -> string not valid. Has to be a number

After:
dim_param: "batch" -> 1

(Pdb) proto
name: "input_ids"
type {
  tensor_type {
    elem_type: 7
    shape {
      dim {
        dim_param: "batch"
      }
      dim {
        dim_value: 2048
      }
    }
  }
}

dim_param: "batch"

dim_value: 1

dim_value: 2048

dim_param: "past_sequence_len + 1"

[1, 1, 2048, 1]