Home
last modified time | relevance | path

Searched refs:create_tensor_like (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2756 return create_tensor_like(tensor_constructor(fill_value))(x, **kwargs)
2806 def create_tensor_like(creation_fn): function
2830 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))