提交 f80ab8fe 编写于 作者: H Haihao Shen

Cast the type to Tensor

上级 9b380f99
......@@ -82,7 +82,7 @@ framework::OpKernelType QuantOp::GetExpectedKernelType(const framework::Executio
layout_ = framework::DataLayout::kMKLDNN;
}
return framework::OpKernelType(
framework::ToDataType(ctx.Input<framework::LoDTensor>("Input")->type()),ctx.GetPlace(),layout_, library_);
framework::ToDataType(ctx.Input<framework::Tensor>("Input")->type()),ctx.GetPlace(),layout_, library_);
//ctx.device_context());
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册