change unique op VisitaDataType from small to tiny (#32872)

PaddlePaddle · May 13, 2021 · b60ab6b · b60ab6b
1 parent eeca963
commit b60ab6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/operators/unique_op.h b/paddle/fluid/operators/unique_op.h
@@ -405,13 +405,13 @@ class UniqueKernel : public framework::OpKernel<T> {
     bool return_counts = context.Attr<bool>("return_counts");
 
     if (axis_vec.empty()) {
-      framework::VisitDataTypeSmall(
+      framework::VisitDataTypeTiny(
           data_type,
           UniqueFlattendTensorFunctor<DeviceContext, T>(
               context, *x, out, return_index, return_inverse, return_counts));
     } else {
       int axis = axis_vec[0];
-      framework::VisitDataTypeSmall(
+      framework::VisitDataTypeTiny(
           data_type, UniqueDimFunctor<DeviceContext, T>(
                          context, *x, out, axis, return_index, return_inverse,
                          return_counts));