Searched refs:get_op_id (Results 1 – 2 of 2) sorted by relevance
271 std::function<int64()> get_op_id) in KernelAndDeviceFunc() argument283 get_op_id_(std::move(get_op_id)) {} in KernelAndDeviceFunc()
523 std::function<int64()> get_op_id = nullptr; in GetOrCreateKernelAndDevice() local525 get_op_id = [&ctx]() { return ctx.RemoteMgr()->NextOpId(); }; in GetOrCreateKernelAndDevice()532 function_outputs_on_op_device, ctx.RendezvousCreator(), get_op_id)); in GetOrCreateKernelAndDevice()