Searched refs:delayed_rewrite_functions (Results 1 – 1 of 1) sorted by relevance
898 forwardprop_input_indices, delayed_rewrite_functions, argument913 self._delayed_rewrite_functions = delayed_rewrite_functions1341 forwardprop_input_indices, delayed_rewrite_functions, argument1345 delayed_rewrite_functions, need_gradients_for_jvps)2147 delayed_rewrite_functions=self._delayed_rewrite_functions,2175 delayed_rewrite_functions=self._delayed_rewrite_functions,