Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Drocm_template.py100 input_tensor_meta=TensorMeta.from_irnodes(self.input_nodes),
101 output_tensor_meta=TensorMeta.from_irnodes(self.output_node),
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_template.py77 input_tensor_meta=TensorMeta.from_irnodes(self.input_nodes),
78 output_tensor_meta=TensorMeta.from_irnodes(self.output_node),
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dcuda_template.py101 input_tensor_meta=TensorMeta.from_irnodes(self.input_nodes),
102 output_tensor_meta=TensorMeta.from_irnodes(self.output_node),
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dautotune_process.py426 def from_irnodes( member in TensorMeta
430 result: List[Any] = [cls.from_irnodes(x) for x in irnodes]
H A Dselect_algorithm.py754 input_tensor_meta=TensorMeta.from_irnodes(full_input_nodes), # type: ignore[arg-type]
755 output_tensor_meta=TensorMeta.from_irnodes(layout),