diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py index 17bf2d544f31dfd749d7583dc75d5fa95b96a0bf..861839256a3da72eb624207a6968be82c14b551d 100755 --- a/python/paddle/__init__.py +++ b/python/paddle/__init__.py @@ -31,6 +31,19 @@ from .fluid.dygraph import monkey_patch_math_varbase monkey_patch_variable() monkey_patch_math_varbase() import paddle.framework +from .framework.dtype import dtype as dtype +from paddle.framework.dtype import uint8 +from paddle.framework.dtype import int8 +from paddle.framework.dtype import int16 +from paddle.framework.dtype import int32 +from paddle.framework.dtype import int64 +from paddle.framework.dtype import float16 +from paddle.framework.dtype import float32 +from paddle.framework.dtype import float64 +from paddle.framework.dtype import bfloat16 +from paddle.framework.dtype import bool +from paddle.framework.dtype import complex64 +from paddle.framework.dtype import complex128 from .framework import VarBase as Tensor import paddle.compat import paddle.distributed diff --git a/python/paddle/framework/dtype.py b/python/paddle/framework/dtype.py new file mode 100644 index 0000000000000000000000000000000000000000..3eeaa6e74ecebd0e76ea11564bbb4060f101c453 --- /dev/null +++ b/python/paddle/framework/dtype.py @@ -0,0 +1,40 @@ +# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +__all__ = [ + "dtype", "uint8", "int8", "int16", "int32", "int64", "bfloat16", "float16", + "float32", "float64", "complex64", "complex128", "bool" +] + +from ..fluid.core import VarDesc + +dtype = VarDesc.VarType +dtype.__qualname__ = "dtype" +dtype.__module__ = "paddle" + +uint8 = VarDesc.VarType.UINT8 +int8 = VarDesc.VarType.INT8 +int16 = VarDesc.VarType.INT16 +int32 = VarDesc.VarType.INT32 +int64 = VarDesc.VarType.INT64 + +float32 = VarDesc.VarType.FP32 +float64 = VarDesc.VarType.FP64 +float16 = VarDesc.VarType.FP16 +bfloat16 = VarDesc.VarType.BF16 + +complex64 = VarDesc.VarType.COMPLEX64 +complex128 = VarDesc.VarType.COMPLEX128 + +bool = VarDesc.VarType.BOOL