Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/extension/llm/custom_ops/
H A Dmodel_sharding.py102 def split_graph(edge_program: ExportedProgram, num_layers: int, shares: int): function
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama3_2/
H A Dllama.py212 model_sharding.split_graph(
/aosp_15_r20/external/executorch/backends/qualcomm/tests/
H A Dutils.py448 def split_graph(self, graph_module: torch.fx.GraphModule, division: int): member in TestQNN
H A Dtest_qnn_delegate.py1527 self.split_graph(edge_prog.exported_program.graph_module, 4)
1845 self.split_graph(edge_prog.exported_program.graph_module, 4)
/aosp_15_r20/external/executorch/examples/models/llama/
H A Dexport_llama_lib.py718 model_sharding.split_graph(