We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3cd99f4 commit 5fe1fe3Copy full SHA for 5fe1fe3
paddle/fluid/operators/softmax_mkldnn_op.cc
@@ -77,7 +77,7 @@ class SoftmaxMKLDNNKernel : public paddle::framework::OpKernel<T> {
77
const bool is_test = ctx.Attr<bool>("is_test");
78
if (!is_test) {
79
T threshold = exp(-64);
80
- for (size_t i = 0; i < dst_tz[0] * dst_tz[1]; ++i) {
+ for (int i = 0; i < dst_tz[0] * dst_tz[1]; ++i) {
81
output_data[i] =
82
output_data[i] < threshold ? threshold : output_data[i];
83
}
0 commit comments