Searched refs:unwrap_tensor (Results 1 – 2 of 2) sorted by relevance
37 def unwrap_tensor(name: str, cur_level_var: str) -> list[str]: function66 unwraps += unwrap_tensor(tensor, cur_level_var)
1426 def unwrap_tensor(x: Union[TensorBox, ir.StorageBox]) -> ir.IRNode: function1443 return can_fuse_reduction(unwrap_tensor(t))1464 return should_lower_cat_input(unwrap_tensor(x))1478 return op_count(unwrap_tensor(x))