54a242de6e148d0836c4a58308b07bbe79c274b6,tensorflow_hub/native_module.py,_ModuleImpl,create_apply_graph,#_ModuleImpl#Any#Any#Any#,472

Before Change

      // infeeds which no longer exist.
      meta_graph_lib.prune_feed_map(meta_graph, infeed_map)
    elif apply_graph.building_function:
      raise NotImplementedError(
          "Using TF-Hub module within a TensorFlow defined function "
          "is currently not supported.")

    // As state ops in the apply graph are unused, replace them with Placeholders
    // so that in a heirarchical instantiation, apply_graph state ops are
    // ignored.

After Change

      //
      // E.g. it could work with "tf.compat.v1.wrap_function", but it will not
      // work with defun, Dataset.map_fn, etc...
      logging.warning("Using `hub.Module` while building a function: %s. This "
                      "can lead to errors if the function is not pruned.",
                      apply_graph.name)

    // As state ops in the apply graph are unused, replace them with Placeholders
    // so that in a heirarchical instantiation, apply_graph state ops are
    // ignored.

Instances