The LazyTensor Datastructures #173
Conversation
| } | ||
| } | ||
|
|
||
| public class LazyTensorOperation : TensorOperation { |
There was a problem hiding this comment.
Is there a reason this type needs to be a public API?
There was a problem hiding this comment.
Actually no. Removed public from class and all APIs.
| public func evaluate() -> [LazyTensor] { | ||
| return Array((0..<outputCount).map { | ||
| LazyTensor(_lazyLive: self, index: $0) | ||
| }) |
| public func evaluate() -> [LazyTensor] { | ||
| return Array((0..<outputCount).map { | ||
| LazyTensor(_lazyLive: self, index: $0) | ||
| }) |
| } | ||
| } | ||
|
|
||
| public class LazyTensorOperation : TensorOperation { |
There was a problem hiding this comment.
Actually no. Removed public from class and all APIs.
|
@rxwei and @dan-zheng, are you OK with this PR? |
| typealias CTFTensor = OpaquePointer | ||
|
|
||
| @usableFromInline | ||
| class LazyTensor : _AnyTensorHandle { |
There was a problem hiding this comment.
| class LazyTensor : _AnyTensorHandle { | |
| class LazyTensor: _AnyTensorHandle { |
| import CTensorFlow | ||
|
|
||
| /// The `TF_Tensor *` type. | ||
| typealias CTFTensor = OpaquePointer |
There was a problem hiding this comment.
Actually, this is an artifact of some earlier version. I removed this altogether.
| case Input.single(let lazyTensor): | ||
| return "\(lazyTensor)" | ||
| case Input.list(let lazyTensorList): do { | ||
| let lazyTensors = lazyTensorList.map{ "\($0)" } |
There was a problem hiding this comment.
The indentation here seems incorrect. Also there should be a space before a trailing closure.
| let lazyTensors = lazyTensorList.map{ "\($0)" } | |
| let lazyTensors = lazyTensorList.map { "\($0)" } |
| @@ -0,0 +1,548 @@ | |||
| import CTensorFlow | |||
There was a problem hiding this comment.
This missed a license header, fixed in #209.
| case .concrete(let h, _): | ||
| return h | ||
| case .symbolic(let op, let index, _): | ||
| assert(false, "TODO: to be send out in a separate PR.") |
There was a problem hiding this comment.
This should be fatalError instead of assert(false, ...).
First of a series of PRs to implement LazyTensor. This PR introduces the
LazyTensorandLazyTensorOperationtypes and associated tests.