Searched refs:calculate_fused_tensor_size (Results 1 – 1 of 1) sorted by relevance
1383 def calculate_fused_tensor_size(split_node: torch.fx.Node, indices: List[int]) -> int: function1442 split_sections[indices[0]] = calculate_fused_tensor_size(1545 start_fused_size = calculate_fused_tensor_size(1548 end_fused_size = start_fused_size + calculate_fused_tensor_size(