def METHOD_NAME(self):
    test_max_difference = torch_device == "cpu"
    relax_max_difference = True
    self._test_inference_batch_single_identical(
        batch_size=2,
        test_max_difference=test_max_difference,
        relax_max_difference=relax_max_difference,
    )