Searched refs:_validate_compile (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 1860 def _validate_compile(self, fn, arg_fn): member in TestNestedTensor
2042 self._validate_compile(fn, arg_fn=lambda: (values, offsets))
2047 self._validate_compile(fn, arg_fn=lambda: (values, different_offsets))
2060 guards_exported, guards_failed = self._validate_compile(
2068 new_guards_exported, _ = self._validate_compile(
2076 self._validate_compile(fn, arg_fn=lambda: (values, offsets3, offsets3))
2084 self._validate_compile(fn, arg_fn=lambda: (values, offsets, offsets))
2100 self._validate_compile(fn, arg_fn=arg_fn)
2112 self._validate_compile(fn, arg_fn=arg_fn)
2122 self._validate_compile(fn, arg_fn=lambda: (values, lengths))
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training.py | 565 self._validate_compile(optimizer, metrics, **kwargs)
2627 def _validate_compile(self, optimizer, metrics, **kwargs): member in Model
|