| 60 | } |
| 61 | |
| 62 | void ConvertToBF16Value(const Tensor& in_tensor, const string name, |
| 63 | BundleWriter& writer) { |
| 64 | auto in_data = in_tensor.flat<float>(); |
| 65 | Tensor out_tensor(DataTypeToEnum<bfloat16>::v(), in_tensor.shape()); |
| 66 | auto out_data = out_tensor.flat<bfloat16>(); |
| 67 | int64 data_size = out_tensor.NumElements(); |
| 68 | #if defined(INTEL_MKL) && !defined(DNNL_AARCH64_USE_ACL) |
| 69 | dnnl::cvt_float_to_bfloat16((dnnl_bfloat16_t*)out_data.data(), |
| 70 | (const float*)in_data.data(), data_size); |
| 71 | #else |
| 72 | FloatToBFloat16(in_data.data(), out_data.data(), data_size); |
| 73 | #endif // INTEL_MKL |
| 74 | writer.Add(name, out_tensor); |
| 75 | } |
| 76 | |
| 77 | void ConvertToHalfValue(const Tensor& in_tensor, const string name, |
| 78 | BundleWriter& writer) { |
no test coverage detected