File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -246,7 +246,7 @@ class CustomOperator : public OperatorWithKernel {
246246 * it can only be determined at runtime.
247247 */
248248 framework::OpKernelType GetExpectedKernelType (
249- const framework::ExecutionContext& ctx) const {
249+ const framework::ExecutionContext& ctx) const override {
250250 return framework::OpKernelType (proto::VarType::RAW, ctx.GetPlace ());
251251 }
252252
@@ -257,7 +257,7 @@ class CustomOperator : public OperatorWithKernel {
257257 */
258258 framework::OpKernelType GetKernelTypeForVar (
259259 const std::string& var_name, const Tensor& tensor,
260- const OpKernelType& expected_kernel_type) {
260+ const OpKernelType& expected_kernel_type) const override {
261261 return OpKernelType (expected_kernel_type.data_type_ ,
262262 expected_kernel_type.place_ , tensor.layout ());
263263 }
You can’t perform that action at this time.
0 commit comments