Skip to content

GitLab

Commit d4a8a1fb authored by Benjamin Thomas Graham's avatar Benjamin Thomas Graham
Browse files

fix leaky relu header_...

parent 9c865087
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
PyTorch/sparseconvnet/SCN/header_cpu.h
View file @ d4a8a1fb
...@@ -216,11 +216,11 @@ void scn_cpu_float_BatchNormalizationInTensor_updateOutput( ...@@ -216,11 +216,11 @@ void scn_cpu_float_BatchNormalizationInTensor_updateOutput(
// LeakyReLU // LeakyReLU
void scn_cpu_float_LeakyReLU_updateOutput( void scn_cpu_float_LeakyReLU_updateOutput(
THFloatTensor *input_features, THFloatTensor *output_features, long n, THFloatTensor *input_features, THFloatTensor *output_features,
float alpha); float alpha);
void scn_cpu_float_LeakyReLU_updateGradInput( void scn_cpu_float_LeakyReLU_updateGradInput(
THFloatTensor *input_features, THFloatTensor *d_input_features, THFloatTensor *input_features, THFloatTensor *d_input_features,
THFloatTensor *d_output_features, long n, float alpha); THFloatTensor *d_output_features, float alpha);
// NetworkInNetwork // NetworkInNetwork
double scn_cpu_float_NetworkInNetwork_updateOutput( double scn_cpu_float_NetworkInNetwork_updateOutput(
......
PyTorch/sparseconvnet/SCN/header_gpu.h
View file @ d4a8a1fb
...@@ -42,11 +42,11 @@ void scn_gpu_float_BatchNormalizationInTensor_updateOutput( ...@@ -42,11 +42,11 @@ void scn_gpu_float_BatchNormalizationInTensor_updateOutput(
// LeakyReLU // LeakyReLU
void scn_gpu_float_LeakyReLU_updateOutput( void scn_gpu_float_LeakyReLU_updateOutput(
THCudaTensor *input_features, THCudaTensor *output_features, long n, THCudaTensor *input_features, THCudaTensor *output_features,
float alpha); float alpha);
void scn_gpu_float_LeakyReLU_updateGradInput( void scn_gpu_float_LeakyReLU_updateGradInput(
THCudaTensor *input_features, THCudaTensor *d_input_features, THCudaTensor *input_features, THCudaTensor *d_input_features,
THCudaTensor *d_output_features, long n, float alpha); THCudaTensor *d_output_features, float alpha);
// NetworkInNetwork // NetworkInNetwork
double scn_gpu_float_NetworkInNetwork_updateOutput( double scn_gpu_float_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