Skip to content

GitLab

Commit 710c4ff0 authored by Benjamin Thomas Graham's avatar Benjamin Thomas Graham
Browse files

fix leaky relu header_...

parent 209e1e78
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
Torch/C.lua
View file @ 710c4ff0
...@@ -108,11 +108,11 @@ void scn_ARCH_REAL_BatchNormalizationInTensor_updateOutput( ...@@ -108,11 +108,11 @@ void scn_ARCH_REAL_BatchNormalizationInTensor_updateOutput(
// LeakyReLU // LeakyReLU
void scn_ARCH_REAL_LeakyReLU_updateOutput( void scn_ARCH_REAL_LeakyReLU_updateOutput(
THTensor *input_features, THTensor *output_features, long n, THTensor *input_features, THTensor *output_features,
float alpha); float alpha);
void scn_ARCH_REAL_LeakyReLU_updateGradInput( void scn_ARCH_REAL_LeakyReLU_updateGradInput(
THTensor *input_features, THTensor *d_input_features, THTensor *input_features, THTensor *d_input_features,
THTensor *d_output_features, long n, float alpha); THTensor *d_output_features, float alpha);
// NetworkInNetwork // NetworkInNetwork
double scn_ARCH_REAL_NetworkInNetwork_updateOutput( double scn_ARCH_REAL_NetworkInNetwork_updateOutput(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment