message: Use `PartialTensorShape` instead of `TensorShape`.
  `TensorShape` constructor throws a CHECK-fail if shape is partial/overflows which the other doesn't. We are only determining the number of elements in the shape and partial shape should be used as it returns negative number when needed.
diff: @@ -45,7 +45,7 @@ constexpr int kMaxTensorNestDepth = 100;
  // not fully defined return -1.
  int64 TensorByteSize(const TensorProto& t) {
  // num_elements returns -1 if shape is not fully defined.
  - int64 num_elems = TensorShape(t.tensor_shape()).num_elements();
  + int64 num_elems = PartialTensorShape(t.tensor_shape()).num_elements();
  return num_elems < 0 ? -1 : num_elems * DataTypeSize(t.dtype());
  }
  