Searched refs:device_branch_fns (Results 1 – 2 of 2) sorted by relevance
77 …argspec: "args=[\'device_branch_fns\', \'default_fn\', \'name\'], varargs=None, keywords=None, def…
3657 def execute_fn_for_device(device_branch_fns, default_fn, name="execute_fn"): argument3693 device_branch_fns_upper = {k.upper(): v for k, v in device_branch_fns.items()}