@@ -750,8 +750,8 @@ extension LazyTensorOperation {
750
750
private func maybeMaterializeInputs( ) {
751
751
/// If `lazyTensor` is symbolic and the associated `LazyTensorOperation`
752
752
/// has been materialized, return the corresponding concrete `LazyTensor`.
753
- /// Otherwise, return `lazyTensor` untouched.
754
- func maybeMaterialized ( lazyTensor: LazyTensor ) -> LazyTensor {
753
+ /// Otherwise, return `lazyTensor` untouched.
754
+ func materializedAsNeeded ( lazyTensor: LazyTensor ) -> LazyTensor {
755
755
let handle = lazyTensor. handle
756
756
if case let LazyTensor . Handle . symbolic( lazyOp, index, _) = handle,
757
757
let outputs = lazyOp. outputs {
@@ -764,15 +764,15 @@ extension LazyTensorOperation {
764
764
/// materialized have been replaced by the corresponding concerete
765
765
/// inputs. If no symbolic values have been materialized or if there are
766
766
/// no symbolic values, return the `input` untouched.
767
- func maybeMaterialized ( input: Input ) -> Input {
767
+ func materializedAsNeeded ( input: Input ) -> Input {
768
768
switch input {
769
769
case . single( let h) :
770
- return . single( maybeMaterialized ( lazyTensor: h) )
770
+ return . single( materializedAsNeeded ( lazyTensor: h) )
771
771
case . list( let elements) :
772
- return . list( elements. map { maybeMaterialized ( lazyTensor: $0) } )
772
+ return . list( elements. map { materializedAsNeeded ( lazyTensor: $0) } )
773
773
}
774
774
}
775
- inputs = inputs. map { maybeMaterialized ( input: $0) }
775
+ inputs = inputs. map { materializedAsNeeded ( input: $0) }
776
776
}
777
777
778
778
private func materializeLiveTensors( ) {
0 commit comments