Searched refs:TRAIN_OP_KEY (Results 1 – 4 of 4) sorted by relevance
60 TRAIN_OP_KEY = "saved_model_train_op" variable
219 train_op = _get_op_from_collection(meta_graph_def, constants.TRAIN_OP_KEY)
511 ops.add_to_collection(constants.TRAIN_OP_KEY, train_op)
1006 self.assertFalse(ops.get_collection(constants.TRAIN_OP_KEY))