Home
last modified time | relevance | path

Searched refs:outer_outputs (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dwrapper.py1932 def codegen_subgraph_prefix(self, subgraph, outer_inputs, outer_outputs): argument
1936 def codegen_subgraph_suffix(self, subgraph, outer_inputs, outer_outputs): argument
1938 subgraph.graph.graph_outputs, outer_outputs
1944 def codegen_subgraph(self, subgraph, outer_inputs, outer_outputs): argument
1948 self.codegen_subgraph_prefix(subgraph, outer_inputs, outer_outputs)
1954 self.codegen_subgraph_suffix(subgraph, outer_inputs, outer_outputs)
1964 outer_outputs = [f"{name}[{i}]" for i in range(len(conditional.outputs))]
1974 self.codegen_subgraph(conditional.true_subgraph, outer_inputs, outer_outputs)
1978 self.codegen_subgraph(conditional.false_subgraph, outer_inputs, outer_outputs)
Dcpp_wrapper_cpu.py1843 def codegen_subgraph_prefix(self, subgraph, outer_inputs, outer_outputs): argument
1863 def codegen_subgraph_suffix(self, subgraph, outer_inputs, outer_outputs): argument
1865 subgraph.graph.graph_outputs, outer_outputs
1882 outer_outputs = []
1887 outer_outputs.append(out.get_name())
1904 outer_outputs = [f"{name}[{i}]" for i in range(len(conditional.outputs))]
1913 self.codegen_subgraph(conditional.true_subgraph, outer_inputs, outer_outputs)
1917 self.codegen_subgraph(conditional.false_subgraph, outer_inputs, outer_outputs)
/external/pytorch/torch/csrc/jit/passes/
Dgraph_fuser.cpp281 auto outer_outputs = outer->outputs(); in mergeFusionGroups() local
283 inner_to_outer[inner_outputs[i]] = outer_outputs[i]; in mergeFusionGroups()