message: Update shape function to support 3/4/5 NCHW dataformat for bias_add
diff: @@ -228,12 +228,12 @@ Status BiasAddShape(shape_inference::InferenceContext* c) {
  if (s.ok() && data_format == "NCHW") {
  // Merge the length of bias_shape into the third to last dimension
  ShapeHandle first;
  - TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, -3, &first));
  + TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, 1, &first));
  ShapeHandle last;
  - TF_RETURN_IF_ERROR(c->Subshape(input_shape, -2, &last));
  + TF_RETURN_IF_ERROR(c->Subshape(input_shape, 2, &last));
  - DimensionHandle input_bias_dim = c->Dim(input_shape, -3);
  + DimensionHandle input_bias_dim = c->Dim(input_shape, 1);
  DimensionHandle merged_bias_dim;
  TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim));
  ShapeHandle merged_bias = c->Vector(merged_bias_dim);
  