Home
last modified time | relevance | path

Searched refs:TFDataFormatToMklDnnDataFormat (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h1073 inline MklTensorFormat TFDataFormatToMklDnnDataFormat(TensorFormat format) { in TFDataFormatToMklDnnDataFormat() function
1127 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in TFShapeToMklDnnDimsInNCHW()
1160 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCHW()
1177 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCDHW()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_grad_input_ops.cc390 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
H A Dmkl_fused_batch_norm_op.cc823 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
1243 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
H A Dmkl_conv_grad_filter_ops.cc438 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
H A Dmkl_pooling_ops_common.h474 is_pool2d ? TFDataFormatToMklDnnDataFormat(this->data_format_tf_) in MklPoolingOpBase()
H A Dmkl_conv_ops.cc733 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute()