Home
last modified time | relevance | path

Searched refs:MklTensorFormatToMklDnnDataFormat (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_concat_op.cc635 auto src_tf_fmt = MklTensorFormatToMklDnnDataFormat( in Compute()
900 MklTensorFormatToMklDnnDataFormat(input_shapes[k].GetTfDataFormat())); in FindMklCommonFormat()
908 MklTensorFormatToMklDnnDataFormat(input_shapes[0].GetTfDataFormat())); in FindMklCommonFormat()
H A Dmkl_conv_grad_input_ops.cc175 MklTensorFormatToMklDnnDataFormat(convBwdInputDims.tf_fmt); in Setup()
393 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
H A Dmkl_conv_grad_filter_ops.cc198 MklTensorFormatToMklDnnDataFormat(convBwdFilterDims.tf_fmt); in Setup()
440 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
H A Dmkl_aggregate_ops.cc173 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_data_format); in Compute()
H A Dmkl_softmax_op.cc124 auto src_format = MklTensorFormatToMklDnnDataFormat(fwdParams.src_fmt); in Setup()
H A Dmkl_matmul_op_fused.cc177 MklTensorFormatToMklDnnDataFormat(MklTensorFormat::FORMAT_NC); in Compute()
H A Dmkl_conv_ops.cc279 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()
H A Dmkl_relu_op.cc644 MklTensorFormatToMklDnnDataFormat(src_mkl_data_format)); in Compute()
661 MklTensorFormatToMklDnnDataFormat(diff_dst_mkl_data_format)); in Compute()
H A Dmkl_fused_batch_norm_op.cc824 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()
1244 dnn_fmt = MklTensorFormatToMklDnnDataFormat(mkl_tensor_fmt); in Compute()
H A Dmkl_pooling_ops_common.h478 MklTensorFormatToMklDnnDataFormat(this->tensor_format_mkldnn_); in MklPoolingOpBase()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h176 memory::format_tag MklTensorFormatToMklDnnDataFormat(MklTensorFormat format);
474 MklTensorFormatToMklDnnDataFormat(data_.tf_data_format_); in GetTfLayout()
1045 inline memory::format_tag MklTensorFormatToMklDnnDataFormat( in MklTensorFormatToMklDnnDataFormat() function