Searched refs:mkl_tensor_fmt (Results 1 – 1 of 1) sorted by relevance
752 MklTensorFormat mkl_tensor_fmt; in Compute() local756 mkl_tensor_fmt = MklTensorFormat::FORMAT_NCHW; in Compute()759 mkl_tensor_fmt = MklTensorFormat::FORMAT_NHWC; in Compute()762 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()763 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()854 dnn_shape_dst.SetTfLayout(ndims, src_dims, mkl_tensor_fmt); in Compute()1142 MklTensorFormat mkl_tensor_fmt; in Compute() local1146 mkl_tensor_fmt = MklTensorFormat::FORMAT_NCHW; in Compute()1149 mkl_tensor_fmt = MklTensorFormat::FORMAT_NHWC; in Compute()1152 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()[all …]