Searched refs:tensor_constructor (Results 1 – 2 of 2) sorted by relevance
195 tensor_constructor: Optional[Callable] = None256 self._tensor_constructor = tensor_constructor
2756 return create_tensor_like(tensor_constructor(fill_value))(x, **kwargs)2759 def tensor_constructor(fill_value): function2830 return create_tensor_like(tensor_constructor(fill_value))2834 ones_like = create_tensor_like(tensor_constructor(1))2835 zeros_like = create_tensor_like(tensor_constructor(0))2918 return tensor_constructor(fill_value)(size, **kwargs)