Searched refs:init_op_key (Results 1 – 2 of 2) sorted by relevance
166 meta_graph_def_to_load, init_op_key=constants.MAIN_OP_KEY): argument185 if init_op_key in collection_def:186 init_op_list = collection_def[init_op_key].node_list.value190 init_op = ops.get_collection(init_op_key)[0]
486 for init_op_key in (constants.MAIN_OP_KEY, constants.LEGACY_INIT_OP_KEY):487 if ops.get_collection(init_op_key):490 "collection {}.".format(init_op_key))