Searched refs:dnn_fmt (Results 1 – 2 of 2) sorted by relevance
167 memory::format_tag dnn_fmt = memory::format_tag::any; in Compute() local174 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_data_format); in Compute()203 md = memory::desc(src_dims, MklDnnType<T>(), dnn_fmt); in Compute()
751 memory::format_tag dnn_fmt; in Compute() local755 dnn_fmt = memory::format_tag::nchw; in Compute()758 dnn_fmt = memory::format_tag::nhwc; in Compute()763 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()774 : memory::desc(src_dims, MklDnnType<T>(), dnn_fmt); in Compute()1141 memory::format_tag dnn_fmt; in Compute() local1145 dnn_fmt = memory::format_tag::nchw; in Compute()1148 dnn_fmt = memory::format_tag::nhwc; in Compute()1153 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()1175 : memory::desc(src_dims, MklDnnType<T>(), dnn_fmt); in Compute()[all …]