|
|
@ -178,10 +178,11 @@ def fc(input,
|
|
|
|
inputs={"Input": input,
|
|
|
|
inputs={"Input": input,
|
|
|
|
"W": w},
|
|
|
|
"W": w},
|
|
|
|
outputs={"Out": tmp},
|
|
|
|
outputs={"Out": tmp},
|
|
|
|
attrs={"use_mkldnn": use_mkldnn,
|
|
|
|
attrs={
|
|
|
|
"is_test": is_test,
|
|
|
|
"use_mkldnn": use_mkldnn,
|
|
|
|
"bias_attr": bias_attr
|
|
|
|
"is_test": is_test,
|
|
|
|
})
|
|
|
|
"bias_attr": bias_attr
|
|
|
|
|
|
|
|
})
|
|
|
|
return helper.append_activation(tmp)
|
|
|
|
return helper.append_activation(tmp)
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
for input_var, param_attr in helper.iter_inputs_and_params():
|
|
|
|
for input_var, param_attr in helper.iter_inputs_and_params():
|
|
|
|