File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff line change @@ -663,6 +663,10 @@ def c_code_cache_version(self):
663
663
664
664
tensor_from_scalar = TensorFromScalar ()
665
665
666
+ @_vectorize_node .register (TensorFromScalar )
667
+ def vectorize_tensor_from_scalar (op , node , batch_x ):
668
+ return identity (batch_x ).owner
669
+
666
670
667
671
class ScalarFromTensor (COp ):
668
672
__props__ = ()
@@ -2046,6 +2050,8 @@ def register_transfer(fn):
2046
2050
"""Create a duplicate of `a` (with duplicated storage)"""
2047
2051
tensor_copy = Elemwise (ps .identity )
2048
2052
pprint .assign (tensor_copy , printing .IgnorePrinter ())
2053
+ identity = tensor_copy
2054
+ pprint .assign (identity , printing .IgnorePrinter ())
2049
2055
2050
2056
2051
2057
class Default (Op ):
@@ -4603,6 +4609,7 @@ def ix_(*args):
4603
4609
"matrix_transpose" ,
4604
4610
"default" ,
4605
4611
"tensor_copy" ,
4612
+ "identity" ,
4606
4613
"transfer" ,
4607
4614
"alloc" ,
4608
4615
"identity_like" ,
You can’t perform that action at this time.
0 commit comments