From ce1bcc947f5d036dad34fabcc854531cb63cbc25 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Mon, 14 May 2018 23:11:23 +0800 Subject: [PATCH] add op to framework.py --- python/paddle/fluid/framework.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 28e54f5492e..46122635404 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -489,7 +489,7 @@ class Operator(object): 'rnn_memory_helper_grad', 'conditional_block', 'while', 'send', 'recv', 'listen_and_serv', 'parallel_do', 'save_combine', 'load_combine', 'ncclInit', 'channel_create', 'channel_close', - 'channel_send', 'channel_recv', 'select' + 'channel_send', 'channel_recv', 'select', 'checkpoint_save' } if type not in no_kernel_op_set: self.desc.infer_var_type(self.block.desc) -- GitLab