Searched refs:_create_fx_call_function (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | sym_node.py | 1126 fx_node, _ = self.shape_env._create_fx_call_function(
1166 fx_node, _ = self.shape_env._create_fx_call_function(op, (self.fx_node,))
1208 fx_node, _ = pred_node.shape_env._create_fx_call_function(
1259 fx_node, _ = self.shape_env._create_fx_call_function(op, tuple(args))
|
| H A D | symbolic_shapes.py | 2871 def _create_fx_call_function( member in ShapeEnv
5163 node, fresh = self._create_fx_call_function(torch._assert, (fx_node,))
5165 neg, _ = self._create_fx_call_function(operator.not_, (fx_node,))
5166 node, fresh = self._create_fx_call_function(torch._assert, (neg,))
5168 eql, _ = self._create_fx_call_function(operator.eq, (fx_node, concrete_val))
5169 node, fresh = self._create_fx_call_function(torch._assert, (eql,))
5375 node, fresh = self._create_fx_call_function(torch._assert, (fx_node,))
|