Home
last modified time | relevance | path

Searched defs:make_prediction (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/functorch/examples/lennard_jones/
H A Dlennard_jones.py50 def make_prediction(model, drs): function
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_freezing.py2033 def make_prediction(self, x): member in TestFreezing.test_freeze_module_with_call_method.Mod
2142 def make_prediction(self, x): member in TestFrozenOptimizations.test_conv_bn_folding_not_forward.ConvBN
3055 def make_prediction(self, x): member in TestFrozenOptimizations.test_freeze_conv_relu_fusion_not_forward.Net
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_eager_transforms.py4327 def make_prediction(model, drs, use_functorch): function