message: Changed to use malloc method from TFLCommonUtils
diff: limitations under the License.
  ==============================================================================*/
  #import "tensorflow_lite_support/ios/task/processor/sources/TFLSegmentationResult.h"
  +#import "tensorflow_lite_support/ios/sources/TFLCommonUtils.h"
  @implementation TFLCategoryMask
  _width = width;
  _height = height;
  if (mask != NULL) {
  - _mask = malloc(width * height * sizeof(UInt8));
  + _mask = [TFLCommonUtils mallocWithSize:width * height * sizeof(UInt8) error:nil];
  + if (_mask) {
  memcpy(_mask, mask, width * height * sizeof(UInt8));
  }
  }
  + }
  return self;
  }
  _width = width;
  _height = height;
  if (mask != NULL) {
  - _mask = malloc(width * height * sizeof(float));
  + _mask = [TFLCommonUtils mallocWithSize:width * height * sizeof(float) error:nil];
  + if (_mask) {
  memcpy(_mask, mask, width * height * sizeof(float));
  }
  }
  + }
  return self;
  }
  