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

Merge pull request #703 from qnqinan/develop

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