make materializationcallback a public var.

bgogul · bgogul · commit 2005d6b35c59 · 2019-06-05T10:32:58.000-07:00
diff --git a/Sources/TensorFlow/LazyTensor/LazyTensorOperation.swift b/Sources/TensorFlow/LazyTensor/LazyTensorOperation.swift
@@ -46,13 +46,8 @@ public class LazyTensor : _AnyTensorHandle {
         handle = Handle.symbolic(op, index: index, isLive: true)
     }
 
-    static var materializationCallback: (String) -> () = {
+    public static var _materializationCallback: (String) -> () = {
         (s: String) in return }
-    
-    public static func registerMaterializationCallback(
-        f: @escaping (String) -> ()) {
-        materializationCallback = f
-    }
 }
 
 class LazyTensorOperation : TensorOperation {

Original file line number	Diff line number	Diff line change
`@@ -46,13 +46,8 @@ public class LazyTensor : _AnyTensorHandle {`
`46`	`46`	`handle = Handle.symbolic(op, index: index, isLive: true)`
`47`	`47`	`}`
`48`	`48`
`49`		`- static var materializationCallback: (String) -> () = {`
	`49`	`+ public static var _materializationCallback: (String) -> () = {`
`50`	`50`	`(s: String) in return }`
`51`		`-`
`52`		`- public static func registerMaterializationCallback(`
`53`		`- f: @escaping (String) -> ()) {`
`54`		`- materializationCallback = f`
`55`		`- }`
`56`	`51`	`}`
`57`	`52`
`58`	`53`	`class LazyTensorOperation : TensorOperation {`