-
Notifications
You must be signed in to change notification settings - Fork 2k
Description
Unsupported for BFloat16 in CPU environment in the browser
System information
Have I written custom code (as opposed to using a stock example script provided in TensorFlow.js): yes
OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Windows 10
TensorFlow.js installed from (npm or script link): script link
TensorFlow.js version (use command below): 3.19.0
Browser version: Microsoft Edge 125.0.2535.67(64 bit)、Google Chrome 125.0.6422.112 (64 bit)
Tensorflow.js Converter Version: -
Exception Report
{
"model_inf": "from:0 to:1 operator:identity from:1 to:2 operator:identity from:1 to:8 operator:batch_normalization from:2 to:3 operator:identity from:3 to:4 operator:identity from:4 to:5 operator:identity from:5 to:6 operator:identity from:6 to:7 operator:identity from:7 to:8 operator:identity from:8 to:9 operator:reducemean ",
"error_message": "Got unsupported ScalarType BFloat16"
}
Bug Description
Such bugs can be reproduced in 8 operators(e.g., BatchNorm, ReduceMean, etc) It seems that certain TensorFlow.js operators does not support BFloat16 on CPU when executing in the browser. Is it? Why?