Searched refs:execute_fn_for_device (Results 1 – 5 of 5) sorted by relevance
76 name: "execute_fn_for_device"
812 runtime) = control_flow_ops.execute_fn_for_device(1638 runtime) = control_flow_ops.execute_fn_for_device(
1280 return control_flow_ops.execute_fn_for_device(branches, lambda: cpu_fn(a))1332 return control_flow_ops.execute_fn_for_device(branches, lambda: cpu_fn(a))1349 return control_flow_ops.execute_fn_for_device(
3657 def execute_fn_for_device(device_branch_fns, default_fn, name="execute_fn"): function
1072 …* Add `execute_fn_for_device function` to dynamically choose the implementation based on underlyin…