Example #1
0
 def typ_apply(self, inputs):
     return core.TensorType(inputs[0].dtype, 4)
Example #2
0
 def typ_apply(self, inputs):
     return core.TupleType(core.TensorType(inputs[0].dtype, 4),
                           core.TensorType('i4', 4))
Example #3
0
 def typ_apply(self, inputs):
     assert inputs[0].ndim == 4
     return core.TensorType(inputs[0].dtype, 4)
Example #4
0
 def typ_apply(self, inputs):
     assert inputs[0].dtype == 'f4'
     d = inputs[0].ndim
     return core.TupleType(core.TensorType(cgt.floatX, d),
                           core.TensorType(cgt.floatX, d))
Example #5
0
 def typ_apply(self, inputs):
     ndim = inputs[0].ndim
     return core.TupleType(core.TensorType(cgt.floatX, ndim),
                           core.TensorType(cgt.floatX, ndim))
Example #6
0
 def typ_apply(self, _inputs):
     return core.TensorType(cgt.floatX, 4)