↓ 7 callersFunctionoptimize_module(
model,
dummy_inputs,
optimize_config=None,
enable_disc=True,
enable_amp=True,
en
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/optimize.py:101
↓ 6 callersMethod_test_factory_like(self, func, d0: int, d1: int, d_or_val: int, numeric_checkable=True, expect_str=None)
pytorch_blade/tests/disc/ops/test_factory_like.py:21