Searched refs:decompose_stack (Results 1 – 1 of 1) sorted by relevance
137 def decompose_stack(graph: torch.fx.GraphModule, input_tensors: List[Any]) -> Any: function245 fused_inputs = decompose_stack(graph, batch_inputs)246 fused_weights = decompose_stack(graph, batch_weights)455 stack_inputs = decompose_stack(graph, batch_inputs)456 stack_others = decompose_stack(graph, batch_others)1024 stack_inputs = decompose_stack(graph, batch_inputs)