Dtype* top_data = (*top)[0]->mutable_gpu_data();
switch (op_) {
case EltwiseParameter_EltwiseOp_PROD:
- caffe_gpu_mul(count, bottom[0]->gpu_data(), bottom[1]->gpu_data(), top_data);
+ caffe_gpu_mul(count, bottom[0]->gpu_data(),
+ bottom[1]->gpu_data(), top_data);
for (int i = 2; i < bottom.size(); ++i) {
caffe_gpu_mul(count, top_data, bottom[i]->gpu_data(), top_data);
}
SUM = 1;
}
optional EltwiseOp operation = 1 [default = SUM]; // element-wise operation
- repeated float coeff = 2; // blob-wise coefficient
+ repeated float coeff = 2; // blob-wise coefficient for SUM operation
}
// Message that stores parameters used by HDF5DataLayer