提交 92f5be1d 编写于 作者: P phlrain

remove inputvarname in operator; test=develop

上级 cf1fe610
...@@ -174,14 +174,6 @@ class ExecutionContext { ...@@ -174,14 +174,6 @@ class ExecutionContext {
return op_.Inputs(name).size(); return op_.Inputs(name).size();
} }
const std::string InputVarName(const std::string& name) const {
return op_.Input(name);
}
const std::string OutputVarName(const std::string& name) const {
return op_.Output(name);
}
size_t OutputSize(const std::string& name) const { size_t OutputSize(const std::string& name) const {
return op_.Outputs(name).size(); return op_.Outputs(name).size();
} }
......
...@@ -292,7 +292,7 @@ class CudnnLSTMGPUKernel : public framework::OpKernel<T> { ...@@ -292,7 +292,7 @@ class CudnnLSTMGPUKernel : public framework::OpKernel<T> {
// multi-devices before the first running. // multi-devices before the first running.
// use parent scope to make cache persistable // use parent scope to make cache persistable
auto *scope = const_cast<framework::Scope *>(ctx.scope().parent()); auto *scope = const_cast<framework::Scope *>(ctx.scope().parent());
auto cache_var_name = ctx.InputVarName("Cache"); auto cache_var_name = ctx.Inputs("Cache")[0];
cache_var = scope->Var(cache_var_name); cache_var = scope->Var(cache_var_name);
} }
CudnnRNNCache *cudnn_rnn_cache = nullptr; CudnnRNNCache *cudnn_rnn_cache = nullptr;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册