提交 c983ca0c 编写于 作者: qnqinan's avatar qnqinan

fix a bug in feed kernel

上级 f565eb45
...@@ -28,8 +28,8 @@ template <> ...@@ -28,8 +28,8 @@ template <>
void FeedKernel<FPGA, float>::Compute(const FeedParam<FPGA> &param) { void FeedKernel<FPGA, float>::Compute(const FeedParam<FPGA> &param) {
auto input = auto input =
reinterpret_cast<Tensor *>(const_cast<LoDTensor *>(param.InputX())); reinterpret_cast<Tensor *>(const_cast<LoDTensor *>(param.InputX()));
auto input_ptr = input->data<float>();
fpga::format_image(input); fpga::format_image(input);
auto input_ptr = input->data<float>();
Tensor *output = param.Out(); Tensor *output = param.Out();
auto output_ptr = output->data<float>(); auto output_ptr = output->data<float>();
......
...@@ -29,8 +29,8 @@ template <> ...@@ -29,8 +29,8 @@ template <>
void FeedKernel<FPGA, float>::Compute(const FeedParam<FPGA> &param) { void FeedKernel<FPGA, float>::Compute(const FeedParam<FPGA> &param) {
auto input = auto input =
reinterpret_cast<Tensor *>(const_cast<LoDTensor *>(param.InputX())); reinterpret_cast<Tensor *>(const_cast<LoDTensor *>(param.InputX()));
auto input_ptr = input->data<float>();
fpga::format_image(input); fpga::format_image(input);
auto input_ptr = input->data<float>();
Tensor *output = param.Out(); Tensor *output = param.Out();
auto output_ptr = output->data<float>(); auto output_ptr = output->data<float>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册