Searched refs:MklTensorFormatToMklDnnDataFormat (Results 1 – 12 of 12) sorted by relevance
52 #define GET_TENSOR_FORMAT(fmt) MklTensorFormatToMklDnnDataFormat(fmt)
174 memory::format_tag MklTensorFormatToMklDnnDataFormat(MklTensorFormat format);476 MklTensorFormatToMklDnnDataFormat(data_.tf_data_format_); in GetTfLayout()1008 inline memory::format_tag MklTensorFormatToMklDnnDataFormat( in MklTensorFormatToMklDnnDataFormat() function
185 MklTensorFormatToMklDnnDataFormat(convBwdInputDims.tf_fmt); in Setup()392 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
623 auto src_tf_fmt = MklTensorFormatToMklDnnDataFormat( in Compute()883 MklTensorFormatToMklDnnDataFormat(input_shapes[k].GetTfDataFormat())); in FindMklCommonFormat()891 MklTensorFormatToMklDnnDataFormat(input_shapes[0].GetTfDataFormat())); in FindMklCommonFormat()
174 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_data_format); in Compute()
116 auto src_format = MklTensorFormatToMklDnnDataFormat(fwdParams.src_fmt); in Setup()
163 MklTensorFormatToMklDnnDataFormat(MklTensorFormat::FORMAT_NC); in Compute()
208 MklTensorFormatToMklDnnDataFormat(convBwdFilterDims.tf_fmt); in Setup()436 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
642 MklTensorFormatToMklDnnDataFormat(src_mkl_data_format)); in Compute()659 MklTensorFormatToMklDnnDataFormat(diff_dst_mkl_data_format)); in Compute()
212 user_data_fmt = MklTensorFormatToMklDnnDataFormat(convFwdDims.tf_fmt); in Setup()599 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()897 auto output_format_tag = MklTensorFormatToMklDnnDataFormat( in AllocateOutputTensor()
463 MklTensorFormatToMklDnnDataFormat(this->tensor_format_mkldnn_); in MklPoolingOpBase()
763 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()1153 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()