@@ -587,27 +587,27 @@ there is only one program, it will by default be saved to the name "forward".
587587
588588 .. code-block :: none
589589
590- Edge Dialect graph of encode
590+ Edge Dialect graph of decode
591591 ExportedProgram:
592592 class GraphModule(torch.nn.Module):
593- def forward(self, x: "f32[1, 10 ]"):
594- # File: /pytorch/executorch/docs/source/tutorials_source/export-to-executorch-tutorial.py:255 in forward, code: return torch.nn.functional.linear(x, torch.randn(5, 10 ))
595- aten_randn_default: "f32[5, 10 ]" = executorch_exir_dialects_edge__ops_aten_randn_default([5, 10 ], device = device(type='cpu'), pin_memory = False)
596- aten_permute_copy_default: "f32[10, 5 ]" = executorch_exir_dialects_edge__ops_aten_permute_copy_default(aten_randn_default, [1, 0]); aten_randn_default = None
597- aten_mm_default: "f32[1, 5 ]" = executorch_exir_dialects_edge__ops_aten_mm_default(x, aten_permute_copy_default); x = aten_permute_copy_default = None
593+ def forward(self, x: "f32[1, 5 ]"):
594+ # File: /pytorch/executorch/docs/source/tutorials_source/export-to-executorch-tutorial.py:260 in forward, code: return torch.nn.functional.linear(x, torch.randn(10, 5 ))
595+ aten_randn_default: "f32[10, 5 ]" = executorch_exir_dialects_edge__ops_aten_randn_default([10, 5 ], device = device(type='cpu'), pin_memory = False)
596+ aten_permute_copy_default: "f32[5, 10 ]" = executorch_exir_dialects_edge__ops_aten_permute_copy_default(aten_randn_default, [1, 0]); aten_randn_default = None
597+ aten_mm_default: "f32[1, 10 ]" = executorch_exir_dialects_edge__ops_aten_mm_default(x, aten_permute_copy_default); x = aten_permute_copy_default = None
598598 return (aten_mm_default,)
599599
600600 Graph signature: ExportGraphSignature(input_specs=[InputSpec(kind=<InputKind.USER_INPUT: 1>, arg=TensorArgument(name='x'), target=None, persistent=None)], output_specs=[OutputSpec(kind=<OutputKind.USER_OUTPUT: 1>, arg=TensorArgument(name='aten_mm_default'), target=None)])
601601 Range constraints: {}
602602
603- Edge Dialect graph of decode
603+ Edge Dialect graph of encode
604604 ExportedProgram:
605605 class GraphModule(torch.nn.Module):
606- def forward(self, x: "f32[1, 5 ]"):
607- # File: /pytorch/executorch/docs/source/tutorials_source/export-to-executorch-tutorial.py:260 in forward, code: return torch.nn.functional.linear(x, torch.randn(10, 5 ))
608- aten_randn_default: "f32[10, 5 ]" = executorch_exir_dialects_edge__ops_aten_randn_default([10, 5 ], device = device(type='cpu'), pin_memory = False)
609- aten_permute_copy_default: "f32[5, 10 ]" = executorch_exir_dialects_edge__ops_aten_permute_copy_default(aten_randn_default, [1, 0]); aten_randn_default = None
610- aten_mm_default: "f32[1, 10 ]" = executorch_exir_dialects_edge__ops_aten_mm_default(x, aten_permute_copy_default); x = aten_permute_copy_default = None
606+ def forward(self, x: "f32[1, 10 ]"):
607+ # File: /pytorch/executorch/docs/source/tutorials_source/export-to-executorch-tutorial.py:255 in forward, code: return torch.nn.functional.linear(x, torch.randn(5, 10 ))
608+ aten_randn_default: "f32[5, 10 ]" = executorch_exir_dialects_edge__ops_aten_randn_default([5, 10 ], device = device(type='cpu'), pin_memory = False)
609+ aten_permute_copy_default: "f32[10, 5 ]" = executorch_exir_dialects_edge__ops_aten_permute_copy_default(aten_randn_default, [1, 0]); aten_randn_default = None
610+ aten_mm_default: "f32[1, 5 ]" = executorch_exir_dialects_edge__ops_aten_mm_default(x, aten_permute_copy_default); x = aten_permute_copy_default = None
611611 return (aten_mm_default,)
612612
613613 Graph signature: ExportGraphSignature(input_specs=[InputSpec(kind=<InputKind.USER_INPUT: 1>, arg=TensorArgument(name='x'), target=None, persistent=None)], output_specs=[OutputSpec(kind=<OutputKind.USER_OUTPUT: 1>, arg=TensorArgument(name='aten_mm_default'), target=None)])
@@ -1270,7 +1270,7 @@ Links Mentioned
12701270
12711271.. rst-class :: sphx-glr-timing
12721272
1273- **Total running time of the script: ** (0 minutes 3.132 seconds)
1273+ **Total running time of the script: ** (0 minutes 3.082 seconds)
12741274
12751275
12761276.. _sphx_glr_download_tutorials_export-to-executorch-tutorial.py :
0 commit comments