diff options
Diffstat (limited to 'reference_model/src/ops')
-rw-r--r-- | reference_model/src/ops/data_layout.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc index f5dbf31..909c567 100644 --- a/reference_model/src/ops/data_layout.cc +++ b/reference_model/src/ops/data_layout.cc @@ -191,6 +191,7 @@ int OpPad<Rank, Dtype>::eval() pad_value = (InEigenType)attribute->pad_const_int(); break; case DType_FP16: + case DType_BF16: case DType_FP32: pad_value = (InEigenType)attribute->pad_const_fp(); break; |