未验证 提交 20dce452 编写于 作者: Z zhangyang0701 提交者: GitHub

Merge pull request #703 from qnqinan/develop

update tensor.h file closed #702
...@@ -257,10 +257,12 @@ class Tensor { ...@@ -257,10 +257,12 @@ class Tensor {
struct FPGAArgs { struct FPGAArgs {
float scale; float scale;
inline const float *scale_pointer() const { return &scale; } inline float *scale_pointer() { return &scale; }
}; };
const struct FPGAArgs fpga_args() const { return fpgaArgs_; } struct FPGAArgs fpga_args() const {
return fpgaArgs_;
}
#endif #endif
private: private:
......
...@@ -41,7 +41,6 @@ bool FusionFcReluKernel<FPGA, float>::Init(FusionFcReluParam *param) { ...@@ -41,7 +41,6 @@ bool FusionFcReluKernel<FPGA, float>::Init(FusionFcReluParam *param) {
fpga::ConvArgs convArgs; fpga::ConvArgs convArgs;
convArgs.relu_enabled = relu_enabled; convArgs.relu_enabled = relu_enabled;
convArgs.bias_address = (void *)input_z_ptr;
convArgs.filter_address = (void *)input_y_ptr; convArgs.filter_address = (void *)input_y_ptr;
convArgs.filter_num = out->dims()[1]; convArgs.filter_num = out->dims()[1];
convArgs.group_num = 1; convArgs.group_num = 1;
......
...@@ -41,7 +41,6 @@ bool FusionFcKernel<FPGA, float>::Init(FusionFcParam *param) { ...@@ -41,7 +41,6 @@ bool FusionFcKernel<FPGA, float>::Init(FusionFcParam *param) {
fpga::ConvArgs convArgs; fpga::ConvArgs convArgs;
convArgs.relu_enabled = relu_enabled; convArgs.relu_enabled = relu_enabled;
convArgs.bias_address = (void *)input_z_ptr;
convArgs.filter_address = (void *)input_y_ptr; convArgs.filter_address = (void *)input_y_ptr;
convArgs.filter_num = out->dims()[1]; convArgs.filter_num = out->dims()[1];
convArgs.group_num = 1; convArgs.group_num = 1;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册