Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torchgen/
H A Dgen_vmap_plumbing.py37 def unwrap_tensor(name: str, cur_level_var: str) -> list[str]: function
66 unwraps += unwrap_tensor(tensor, cur_level_var)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py1426 def unwrap_tensor(x: Union[TensorBox, ir.StorageBox]) -> ir.IRNode: function
1443 return can_fuse_reduction(unwrap_tensor(t))
1464 return should_lower_cat_input(unwrap_tensor(x))
1478 return op_count(unwrap_tensor(x))