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 0d4a5ed commit 2c63250Copy full SHA for 2c63250
graph_net/paddle/test_compiler.py
@@ -82,9 +82,7 @@ def get_input_spec(args):
82
83
84
def regular_item(item):
85
- if isinstance(item, paddle.Tensor) and (
86
- item.dtype == paddle.bfloat16 or item.dtype == paddle.bfloat32
87
- ):
+ if isinstance(item, paddle.Tensor) and (item.dtype == paddle.bfloat16):
88
item = np.array(item.astype("float32"))
89
else:
90
item = np.array(item)
0 commit comments