Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_export/passes/
H A Dconstant_folding.py170 if not self.insertable_tensor_check(out):
192 def insertable_tensor_check(self, tensor: torch.Tensor) -> bool: member in ConstantFolder
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dconstant_folding.py197 if not self.insertable_tensor_check(out):
219 def insertable_tensor_check(self, tensor: torch.Tensor) -> bool: member in ConstantFolder
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Djoint_graph.py237 def insertable_tensor_check(self, t: torch.Tensor) -> bool: member in UniformValueConstantFolder