Searched refs:init_op_key (Results 1 – 2 of 2) sorted by relevance
135 meta_graph_def_to_load, init_op_key=constants.MAIN_OP_KEY): argument154 if init_op_key in collection_def:155 init_op_list = collection_def[init_op_key].node_list.value159 init_op = ops.get_collection(init_op_key)[0]
476 for init_op_key in (constants.MAIN_OP_KEY, constants.LEGACY_INIT_OP_KEY):477 if ops.get_collection(init_op_key):480 "collection {}.".format(init_op_key))