Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1036 inline MklTensorFormat TFDataFormatToMklDnnDataFormat(TensorFormat format) { in TFDataFormatToMklDnnDataFormat() function
1090 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in TFShapeToMklDnnDimsInNCHW()
1123 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCHW()
1140 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCDHW()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc389 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
Dmkl_conv_grad_filter_ops.cc434 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
Dmkl_pooling_ops_common.h459 is_pool2d ? TFDataFormatToMklDnnDataFormat(this->data_format_tf_) in MklPoolingOpBase()
Dmkl_fused_batch_norm_op.cc762 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
1152 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
Dmkl_conv_ops.cc596 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute()