提交 6c20e7c4 编写于 作者: S silingtong123 提交者: hong

test=develop, remove unused parameter from class RuntimeInferShapeContext constructors (#22046)

上级 fab4b076
...@@ -532,8 +532,7 @@ bool OpSupportGPU(const std::string& op_type) { ...@@ -532,8 +532,7 @@ bool OpSupportGPU(const std::string& op_type) {
class RuntimeInferShapeContext : public InferShapeContext { class RuntimeInferShapeContext : public InferShapeContext {
public: public:
RuntimeInferShapeContext(const OperatorBase& op, const Scope& scope, RuntimeInferShapeContext(const OperatorBase& op, const RuntimeContext& ctx)
const RuntimeContext& ctx)
: op_(op), ctx_(ctx) {} : op_(op), ctx_(ctx) {}
bool HasInput(const std::string& name) const override { bool HasInput(const std::string& name) const override {
...@@ -901,7 +900,7 @@ static void CheckTensorNANOrInf(const std::string& op_type, ...@@ -901,7 +900,7 @@ static void CheckTensorNANOrInf(const std::string& op_type,
void OperatorWithKernel::RuntimeInferShape(const Scope& scope, void OperatorWithKernel::RuntimeInferShape(const Scope& scope,
const platform::Place& place, const platform::Place& place,
const RuntimeContext& ctx) const { const RuntimeContext& ctx) const {
RuntimeInferShapeContext infer_shape_ctx(*this, scope, ctx); RuntimeInferShapeContext infer_shape_ctx(*this, ctx);
this->InferShape(&infer_shape_ctx); this->InferShape(&infer_shape_ctx);
} }
...@@ -966,7 +965,7 @@ void OperatorWithKernel::RunImpl(const Scope& scope, ...@@ -966,7 +965,7 @@ void OperatorWithKernel::RunImpl(const Scope& scope,
} }
if (!all_kernels_must_compute_runtime_shape_) { if (!all_kernels_must_compute_runtime_shape_) {
RuntimeInferShapeContext infer_shape_ctx(*this, exec_scope, *runtime_ctx); RuntimeInferShapeContext infer_shape_ctx(*this, *runtime_ctx);
this->InferShape(&infer_shape_ctx); this->InferShape(&infer_shape_ctx);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册