Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc390 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
Dmkl_conv_grad_filter_ops.cc435 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
Dmkl_pooling_ops_common.h460 : TFDataFormatToMklDnn3DDataFormat(this->data_format_tf_); in MklPoolingOpBase()
Dmkl_conv_ops.cc597 : TFDataFormatToMklDnn3DDataFormat(data_format_); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1024 inline MklTensorFormat TFDataFormatToMklDnn3DDataFormat(TensorFormat format) { in TFDataFormatToMklDnn3DDataFormat() function
1105 DCHECK_NE(TFDataFormatToMklDnn3DDataFormat(format), in TFShapeToMklDnnDimsInNCDHW()