Searched refs:_get_op_from_collection (Results 1 – 1 of 1) sorted by relevance
163 def _get_op_from_collection(meta_graph_def, op_key): function180 _get_op_from_collection(meta_graph_def, constants.MAIN_OP_KEY) or181 _get_op_from_collection(meta_graph_def, constants.LEGACY_INIT_OP_KEY))188 train_op = _get_op_from_collection(meta_graph_def, constants.TRAIN_OP_KEY)