fix-typo
typhoonzero 8 years ago
parent 12f0a86f36
commit bf7bc1276f

@ -33,7 +33,7 @@ class AucOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_EQ(inference->dims(), label->dims(),
"inference and label should have same shape");
ctx.Output<framework::Tensor>("AUC")->Resize({1});
ctx.Output<framework::LoDTensor>("AUC")->Resize({1});
}
};

Loading…
Cancel
Save