Searched refs:variable_input_fn (Results 1 – 1 of 1) sorted by relevance
255 variable_input_fn = lambda unused_variables: _loss() function256 forward_over_back_hvp, = _hvp(variable_input_fn,