Searched refs:create_tensor_like (Results 1 – 1 of 1) sorted by relevance
2756 return create_tensor_like(tensor_constructor(fill_value))(x, **kwargs)2806 def create_tensor_like(creation_fn): function2830 return create_tensor_like(tensor_constructor(fill_value))2833 empty_like = register_lowering(aten.empty_like)(create_tensor_like(empty))2834 ones_like = create_tensor_like(tensor_constructor(1))2835 zeros_like = create_tensor_like(tensor_constructor(0))