Searched defs:assert_fn (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_mixed_precision.py | 80 def assert_fn(output: torch.Tensor): function
149 def assert_fn(output: torch.Tensor): function
193 def assert_fn(output: torch.Tensor): function
264 def assert_fn(output: torch.Tensor): function
|
| H A D | test_fully_shard_frozen.py | 98 def assert_fn(output: torch.Tensor): function
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_state_dict.py | 163 def _state_compare(self, model, model_new, assert_fn, state_generator="parameters"): argument
171 self, model, model_new, assert_fn, check_fp16=False, check_buffers=True argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_dispatch_test.py | 1155 assert_fn = lambda x, y: self.assertAllClose(x, y, rtol=rtol) function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | wrap_function_test.py | 432 def assert_fn(): function
|
| /aosp_15_r20/external/pytorch/torch/fx/ |
| H A D | proxy.py | 356 def assert_fn(x): function
|
| /aosp_15_r20/external/google-styleguide/cpplint/ |
| H A D | cpplint_unittest.py | 60 def __init__(self, assert_fn): argument
|