Searched refs:MklTensorFormatToMklDnnDataFormat (Results 1 – 11 of 11) sorted by relevance
635 auto src_tf_fmt = MklTensorFormatToMklDnnDataFormat( in Compute()900 MklTensorFormatToMklDnnDataFormat(input_shapes[k].GetTfDataFormat())); in FindMklCommonFormat()908 MklTensorFormatToMklDnnDataFormat(input_shapes[0].GetTfDataFormat())); in FindMklCommonFormat()
175 MklTensorFormatToMklDnnDataFormat(convBwdInputDims.tf_fmt); in Setup()393 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
198 MklTensorFormatToMklDnnDataFormat(convBwdFilterDims.tf_fmt); in Setup()440 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
173 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_data_format); in Compute()
124 auto src_format = MklTensorFormatToMklDnnDataFormat(fwdParams.src_fmt); in Setup()
177 MklTensorFormatToMklDnnDataFormat(MklTensorFormat::FORMAT_NC); in Compute()
279 user_data_fmt = MklTensorFormatToMklDnnDataFormat(convFwdDims.tf_fmt); in Setup()316 : MklTensorFormatToMklDnnDataFormat(convFwdDims.tf_fmt); in Setup()736 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()1133 auto output_format_tag = MklTensorFormatToMklDnnDataFormat( in AllocateOutputTensor()
644 MklTensorFormatToMklDnnDataFormat(src_mkl_data_format)); in Compute()661 MklTensorFormatToMklDnnDataFormat(diff_dst_mkl_data_format)); in Compute()
824 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()1244 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()
478 MklTensorFormatToMklDnnDataFormat(this->tensor_format_mkldnn_); in MklPoolingOpBase()
176 memory::format_tag MklTensorFormatToMklDnnDataFormat(MklTensorFormat format);474 MklTensorFormatToMklDnnDataFormat(data_.tf_data_format_); in GetTfLayout()1045 inline memory::format_tag MklTensorFormatToMklDnnDataFormat( in MklTensorFormatToMklDnnDataFormat() function