Function registerClass

tfjs-core/src/serialization.ts:234–265

(
    cls: SerializableConstructor<T>, pkg?: string, name?: string)

 * @doc {heading: 'Models', subheading: 'Serialization', ignoreCI: true}
 */
export function registerClass<T extends Serializable>(
    cls: SerializableConstructor<T>, pkg?: string, name?: string) {
  assert(
      cls.className != null,
      () => `Class being registered does not have the static className ` +
          `property defined.`);
  assert(
      typeof cls.className === 'string',
      () => `className is required to be a string, but got type ` +
          typeof cls.className);
  assert(
      cls.className.length > 0,
      () => `Class being registered has an empty-string as its className, ` +
          `which is disallowed.`);

  if (typeof pkg === 'undefined') {
    pkg = 'Custom';
  }

  if (typeof name === 'undefined') {
    name = cls.className;
  }

  const className = name;
  const registerName = pkg + '>' + className;

  SerializationMap.register(cls);
  GLOBAL_CUSTOM_OBJECT.set(registerName, cls);
  GLOBAL_CUSTOM_NAMES.set(cls, registerName);

  return cls;
}

/**
 * Get the registered name of a class. If the class has not been registered,