|
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