Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_fused_batch_norm_op.cc606 format_m = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
916 format_m = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
Dmkl_conv_grad_input_ops.cc378 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
Dmkl_conv_grad_filter_ops.cc439 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
Dmkl_pooling_ops_common.h474 is_pool2d ? TFDataFormatToMklDnnDataFormat(this->data_format_tf_) in MklPoolingOpBase()
Dmkl_conv_ops.cc1020 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1464 inline memory::format TFDataFormatToMklDnnDataFormat(TensorFormat format) { in TFDataFormatToMklDnnDataFormat() function
1519 CHECK_NE(TFDataFormatToMklDnnDataFormat(format), in TFShapeToMklDnnDimsInNCHW()
1552 CHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCHW()