fix Print_op input dtype list error test=develop (#21326)

revert-21172-masked_select_api
lijianshe02 5 years ago committed by GitHub
parent 84865b806b
commit 7fcaa39b36
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -254,7 +254,7 @@ def Print(input,
'''
check_type_and_dtype(input, 'input', Variable,
['float32', 'float64', 'int32_t', 'int64_t', 'bool'],
['float32', 'float64', 'int32', 'int64', 'bool'],
'fluid.layers.Print')
helper = LayerHelper('print' + "_" + input.name, **locals())

Loading…
Cancel
Save