Searched refs:op_fn (Results 1 – 3 of 3) sorted by relevance
52 ffsb_op_fn op_fn; member
191 ffsb_op_list[op_num].op_fn(ft, fs, op_num); in do_op()
1261 device=host_device, op_fn=enqueue_ops_fn))1281 enqueue_ops.append(wrap_fn(device=host_device, op_fn=enqueue_ops_fn))1322 wrap_fn(device=host_device, op_fn=enqueue_ops_fn))3248 def _wrap_computation_in_while_loop(device, op_fn): argument3252 with ops.control_dependencies(op_fn()):3266 def _wrap_computation_in_while_loop_with_stopping_signals(device, op_fn): argument3274 return_value = op_fn()