21 #ifndef SINGA_UTILS_MKLDNN_UTILS_H_ 22 #define SINGA_UTILS_MKLDNN_UTILS_H_ 35 inline mkldnn::memory::data_type GetMKLDNNDataType(DataType dtype) {
36 mkldnn::memory::data_type ret = mkldnn::memory::data_type::f32;
39 ret = mkldnn::memory::data_type::f32;
42 LOG(FATAL) <<
"The data type " << DataType_Name(dtype)
43 <<
" is not support by mkldnn";
46 LOG(FATAL) <<
"The data type " << DataType_Name(dtype)
47 <<
" is not support by mkldnn";
50 LOG(FATAL) <<
"The data type " << DataType_Name(dtype)
51 <<
" is not support by mkldnn";
56 #endif // SINGA_UTILS_MKLDNN_UTILS_H_ Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements...
Definition: common.h:48