Searched refs:TFDataFormatToMklDnnDataFormat (Results 1 – 6 of 6) sorted by relevance
1073 inline MklTensorFormat TFDataFormatToMklDnnDataFormat(TensorFormat format) { in TFDataFormatToMklDnnDataFormat() function1127 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in TFShapeToMklDnnDimsInNCHW()1160 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCHW()1177 DCHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCDHW()
390 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
823 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()1243 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
438 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
474 is_pool2d ? TFDataFormatToMklDnnDataFormat(this->data_format_tf_) in MklPoolingOpBase()
733 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute()