mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
538d2412ef
This pull request contains a few changes: 1. Adds support for string ort values. 2. Fixes the training minimal build (that was broken with #16601) by putting custom op registration behind #ifdefs 3. Fixes the iOS pod package generation (that was again broken with #16601) by explicitly providing paths to be copied during pod creation.
122 lines
3.7 KiB
Objective-C
122 lines
3.7 KiB
Objective-C
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#import <Foundation/Foundation.h>
|
|
|
|
#import "ort_enums.h"
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
@class ORTValueTypeInfo;
|
|
@class ORTTensorTypeAndShapeInfo;
|
|
|
|
/**
|
|
* An ORT value encapsulates data used as an input or output to a model at runtime.
|
|
*/
|
|
@interface ORTValue : NSObject
|
|
|
|
- (instancetype)init NS_UNAVAILABLE;
|
|
|
|
/**
|
|
* Creates a value that is a tensor.
|
|
* The tensor data is allocated by the caller.
|
|
*
|
|
* @param tensorData The tensor data.
|
|
* @param elementType The tensor element data type.
|
|
* @param shape The tensor shape.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The instance, or nil if an error occurs.
|
|
*/
|
|
- (nullable instancetype)initWithTensorData:(NSMutableData*)tensorData
|
|
elementType:(ORTTensorElementDataType)elementType
|
|
shape:(NSArray<NSNumber*>*)shape
|
|
error:(NSError**)error;
|
|
|
|
/**
|
|
* Creates a value that is a string tensor.
|
|
* The string data will be copied into a buffer owned by this ORTValue instance.
|
|
*
|
|
* Available since 1.16.
|
|
*
|
|
* @param tensorStringData The tensor string data.
|
|
* @param shape The tensor shape.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The instance, or nil if an error occurs.
|
|
*/
|
|
- (nullable instancetype)initWithTensorStringData:(NSArray<NSString*>*)tensorStringData
|
|
shape:(NSArray<NSNumber*>*)shape
|
|
error:(NSError**)error;
|
|
|
|
/**
|
|
* Gets the type information.
|
|
*
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The type information, or nil if an error occurs.
|
|
*/
|
|
- (nullable ORTValueTypeInfo*)typeInfoWithError:(NSError**)error;
|
|
|
|
/**
|
|
* Gets the tensor type and shape information.
|
|
* This assumes that the value is a tensor.
|
|
*
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The tensor type and shape information, or nil if an error occurs.
|
|
*/
|
|
- (nullable ORTTensorTypeAndShapeInfo*)tensorTypeAndShapeInfoWithError:(NSError**)error;
|
|
|
|
/**
|
|
* Gets the tensor data.
|
|
* This assumes that the value is a tensor.
|
|
*
|
|
* This returns the value's underlying data directly, not a copy of it.
|
|
* The memory's lifetime may be tied to this value, i.e., if it was allocated
|
|
* by ORT. On the other hand, the memory's lifetime is independent of the value
|
|
* if the value was created with user-provided data.
|
|
*
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The tensor data, or nil if an error occurs.
|
|
*/
|
|
- (nullable NSMutableData*)tensorDataWithError:(NSError**)error;
|
|
|
|
/**
|
|
* Gets the tensor string data.
|
|
* This assumes that the value is a string tensor.
|
|
*
|
|
* This returns a copy of the value's underlying string data.
|
|
*
|
|
* Available since 1.16.
|
|
*
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The copy of the tensor string data, or nil if an error occurs.
|
|
*/
|
|
- (nullable NSArray<NSString*>*)tensorStringDataWithError:(NSError**)error;
|
|
|
|
@end
|
|
|
|
/**
|
|
* A value's type information.
|
|
*/
|
|
@interface ORTValueTypeInfo : NSObject
|
|
|
|
/** The value type. */
|
|
@property(nonatomic) ORTValueType type;
|
|
|
|
/** The tensor type and shape information, if the value is a tensor. */
|
|
@property(nonatomic, nullable) ORTTensorTypeAndShapeInfo* tensorTypeAndShapeInfo;
|
|
|
|
@end
|
|
|
|
/**
|
|
* A tensor's type and shape information.
|
|
*/
|
|
@interface ORTTensorTypeAndShapeInfo : NSObject
|
|
|
|
/** The tensor element data type. */
|
|
@property(nonatomic) ORTTensorElementDataType elementType;
|
|
|
|
/** The tensor shape. */
|
|
@property(nonatomic) NSArray<NSNumber*>* shape;
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|