Home
last modified time | relevance | path

Searched refs:TFDataFormatToMklDnn3DDataFormat (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_grad_input_ops.cc391 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
H A Dmkl_conv_grad_filter_ops.cc439 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
H A Dmkl_pooling_ops_common.h475 : TFDataFormatToMklDnn3DDataFormat(this->data_format_tf_); in MklPoolingOpBase()
H A Dmkl_conv_ops.cc734 : TFDataFormatToMklDnn3DDataFormat(data_format_); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h1061 inline MklTensorFormat TFDataFormatToMklDnn3DDataFormat(TensorFormat format) { in TFDataFormatToMklDnn3DDataFormat() function
1142 DCHECK_NE(TFDataFormatToMklDnn3DDataFormat(format), in TFShapeToMklDnnDimsInNCDHW()