|
11 | 11 | /// A protocol that types conform to in order to be used as tensor element types. |
12 | 12 | /// Provides the mapping from the Swift type to the underlying `DataType`. |
13 | 13 | @available(*, deprecated, message: "This API is experimental.") |
14 | | -protocol Scalar { |
| 14 | +public protocol Scalar { |
15 | 15 | /// The `DataType` corresponding to this scalar type. |
16 | 16 | static var dataType: DataType { get } |
17 | 17 | } |
18 | 18 |
|
19 | 19 | @available(*, deprecated, message: "This API is experimental.") |
20 | | -extension UInt8: Scalar { static var dataType: DataType { .byte } } |
| 20 | +extension UInt8: Scalar { public static var dataType: DataType { .byte } } |
21 | 21 | @available(*, deprecated, message: "This API is experimental.") |
22 | | -extension Int8: Scalar { static var dataType: DataType { .char } } |
| 22 | +extension Int8: Scalar { public static var dataType: DataType { .char } } |
23 | 23 | @available(*, deprecated, message: "This API is experimental.") |
24 | | -extension Int16: Scalar { static var dataType: DataType { .short } } |
| 24 | +extension Int16: Scalar { public static var dataType: DataType { .short } } |
25 | 25 | @available(*, deprecated, message: "This API is experimental.") |
26 | | -extension Int32: Scalar { static var dataType: DataType { .int } } |
| 26 | +extension Int32: Scalar { public static var dataType: DataType { .int } } |
27 | 27 | @available(*, deprecated, message: "This API is experimental.") |
28 | | -extension Int64: Scalar { static var dataType: DataType { .long } } |
| 28 | +extension Int64: Scalar { public static var dataType: DataType { .long } } |
29 | 29 | @available(*, deprecated, message: "This API is experimental.") |
30 | | -extension Int: Scalar { static var dataType: DataType { .long } } |
| 30 | +extension Int: Scalar { public static var dataType: DataType { .long } } |
31 | 31 | @available(*, deprecated, message: "This API is experimental.") |
32 | | -extension Float: Scalar { static var dataType: DataType { .float } } |
| 32 | +extension Float: Scalar { public static var dataType: DataType { .float } } |
33 | 33 | @available(*, deprecated, message: "This API is experimental.") |
34 | | -extension Double: Scalar { static var dataType: DataType { .double } } |
| 34 | +extension Double: Scalar { public static var dataType: DataType { .double } } |
35 | 35 | @available(*, deprecated, message: "This API is experimental.") |
36 | | -extension Bool: Scalar { static var dataType: DataType { .bool } } |
| 36 | +extension Bool: Scalar { public static var dataType: DataType { .bool } } |
37 | 37 | @available(*, deprecated, message: "This API is experimental.") |
38 | | -extension UInt16: Scalar { static var dataType: DataType { .uInt16 } } |
| 38 | +extension UInt16: Scalar { public static var dataType: DataType { .uInt16 } } |
39 | 39 | @available(*, deprecated, message: "This API is experimental.") |
40 | | -extension UInt32: Scalar { static var dataType: DataType { .uInt32 } } |
| 40 | +extension UInt32: Scalar { public static var dataType: DataType { .uInt32 } } |
41 | 41 | @available(*, deprecated, message: "This API is experimental.") |
42 | | -extension UInt64: Scalar { static var dataType: DataType { .uInt64 } } |
| 42 | +extension UInt64: Scalar { public static var dataType: DataType { .uInt64 } } |
43 | 43 | @available(*, deprecated, message: "This API is experimental.") |
44 | | -extension UInt: Scalar { static var dataType: DataType { .uInt64 } } |
| 44 | +extension UInt: Scalar { public static var dataType: DataType { .uInt64 } } |
45 | 45 |
|
46 | 46 | @available(*, deprecated, message: "This API is experimental.") |
47 | | -extension Tensor { |
| 47 | +public extension Tensor { |
48 | 48 | /// Calls the closure with a typed, immutable buffer pointer over the tensor’s elements. |
49 | 49 | /// |
50 | 50 | /// - Parameter body: A closure that receives an `UnsafeBufferPointer<T>` bound to the tensor’s data. |
|
0 commit comments