| 74 | color_normalization_(1.0f / (2.0f * sigma_color * sigma_color)) {} |
| 75 | |
| 76 | __device__ inline float Compute(const float row_diff, |
| 77 | const float col_diff, |
| 78 | const float color1, |
| 79 | const float color2) const { |
| 80 | const float spatial_dist_squared = |
| 81 | row_diff * row_diff + col_diff * col_diff; |
| 82 | const float color_dist = color1 - color2; |
| 83 | return exp(-spatial_dist_squared * spatial_normalization_ - |
| 84 | color_dist * color_dist * color_normalization_); |
| 85 | } |
| 86 | |
| 87 | private: |
| 88 | const float spatial_normalization_; |