Searched refs:initial_value_fn (Results 1 – 3 of 3) sorted by relevance
263 initial_value_fn = initial_value265 initial_value_fn = lambda: initial_value function276 initial_value = initial_value_fn()
427 def initial_value_fn(device=d): function430 kwargs["initial_value"] = initial_value_fn
544 def initial_value_fn(device=d): function552 kwargs["initial_value"] = initial_value_fn