#include #include "caffe/layers/silence_layer.hpp" #include "caffe/util/math_functions.hpp" namespace caffe { template void SilenceLayer::Backward_cpu(const vector*>& top, const vector& propagate_down, const vector*>& bottom) { for (int i = 0; i < bottom.size(); ++i) { if (propagate_down[i]) { caffe_set(bottom[i]->count(), Dtype(0), bottom[i]->mutable_cpu_diff()); } } } #ifdef CPU_ONLY STUB_GPU(SilenceLayer); #endif INSTANTIATE_CLASS(SilenceLayer); REGISTER_LAYER_CLASS(Silence); } // namespace caffe