def _InitOpDefLibrary(op_list_proto_bytes): op_list = _op_def_pb2.OpList() op_list.ParseFromString(op_list_proto_bytes) _op_def_registry.register_op_list(op_list) op_def_lib = _op_def_library.OpDefLibrary() op_def_lib.add_op_list(op_list) return op_def_lib
def _InitOpDefLibrary(): op_list = _op_def_pb2.OpList() _text_format.Merge(_InitOpDefLibrary.op_list_ascii, op_list) _op_def_registry.register_op_list(op_list) op_def_lib = _op_def_library.OpDefLibrary() op_def_lib.add_op_list(op_list) return op_def_lib
def _create_op_def_library(op_protos): for op_proto in op_protos: registered_ops = _registry.get_registered_ops() if op_proto.name not in registered_ops: raise LookupError("Op with name {0} not registered".format( op_proto.name)) op_def_lib = _op_def_library.OpDefLibrary() ops_proto = _op_def_pb2.OpList() ops_proto.op.extend([op_proto]) # Fails if the interfaces ("op schemas") don't match between the # previously registered op and this one. _registry.register_op_list(ops_proto) op_def_lib.add_op_list(ops_proto) return op_def_lib