Searched refs:use_custom_function (Results 1 – 1 of 1) sorted by relevance
1775 def _get_mul2(self, use_custom_function): argument1776 if use_custom_function:1792 for use_custom_function in (True, False):1793 mul2 = self._get_mul2(use_custom_function)1895 for use_custom_function in (True, False):1896 mul2 = self._get_mul2(use_custom_function)9129 def test_hook_closure_cycle(self, use_custom_function, use_tensor_hook): argument9155 if use_custom_function: