Method Set

tensorflow/core/common_runtime/function.cc:1295–1298

  CustomKernelCreator* custom_creator = nullptr;

  void Set(CustomKernelCreator* cb) {
    mutex_lock l(mu);
    custom_creator = cb;
  }

  CustomKernelCreator* Get() {
    mutex_lock l(mu);