03ff0c8f1cbcb4869e3cce95395c66178c02f576,tensorflow_hub/native_module.py,,list_registered_stateful_ops_without_inputs,#,594

Before Change

  Returns:
    A set of strings.
  
  return set([
      name
      for name, op in op_def_registry.get_registered_ops().items()
      if op.is_stateful and not op.input_arg
  ])


def get_state_map(meta_graph, state_ops, unsupported_state_ops,

After Change

    A set of strings.
  
  used_ops = (node.op for node in graph_def.node)
  return {op for op in used_ops if is_registered_stateful_op_without_inputs(op)}


def get_state_map(meta_graph, state_ops, unsupported_state_ops,
                  get_tensor_by_name):

Instances