Unify `set_feed_variable` to one method (#4949)

revert-4814-Add_sequence_project_op
Yu Yang 8 years ago committed by GitHub
parent c532b96741
commit 42f2dd4041

@ -21,7 +21,6 @@ limitations under the License. */
namespace paddle {
namespace framework {
template <typename T>
void SetFeedVariable(const LoDTensor& input, const std::string& var_name,
size_t index) {
// If var_name Variable is not found in GlobalScope, a new variable will

@ -460,10 +460,7 @@ All parameter, weight, gradient are variables in Paddle.
m.def("unique_integer", UniqueIntegerGenerator);
m.def("is_compile_gpu", IsCompileGPU);
//! FIXME: it is no need to `set_xxx_float/double/int`
m.def("set_feed_variable_float", framework::SetFeedVariable<float>);
m.def("set_feed_variable_double", framework::SetFeedVariable<double>);
m.def("set_feed_variable_int", framework::SetFeedVariable<int>);
m.def("set_feed_variable", framework::SetFeedVariable);
m.def("get_fetch_variable", framework::GetFetchVariable);
BindProgramDesc(m);

@ -38,8 +38,7 @@ class Executor(object):
inputs={'X': [feed_var]},
outputs={'Out': [out]},
attrs={'col': i})
# FIXME
core.set_feed_variable_float(feed[name], feed_var.name, i)
core.set_feed_variable(feed[name], feed_var.name, i)
fetch_var = global_block.create_var(
name=fetch_var_name,

@ -12,7 +12,7 @@ class TestFeedFetch(unittest.TestCase):
input_tensor = core.LoDTensor([[0, 2, 4]])
input_tensor.set(input_array, place)
core.set_feed_variable_float(input_tensor, "feed", 0)
core.set_feed_variable(input_tensor, "feed", 0)
output_tensor = core.get_fetch_variable("feed", 0)

Loading…
Cancel
Save