Searched refs:delayed_rewrite_functions (Results 1 – 1 of 1) sorted by relevance
880 forwardprop_input_indices, delayed_rewrite_functions, argument895 self._delayed_rewrite_functions = delayed_rewrite_functions1325 forwardprop_input_indices, delayed_rewrite_functions, argument1329 delayed_rewrite_functions, need_gradients_for_jvps)2132 delayed_rewrite_functions=self._delayed_rewrite_functions,2160 delayed_rewrite_functions=self._delayed_rewrite_functions,