Searched refs:TFDataFormatToMklDnnDataFormat (Results 1 – 6 of 6) sorted by relevance
606 format_m = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()916 format_m = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
378 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
439 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
474 is_pool2d ? TFDataFormatToMklDnnDataFormat(this->data_format_tf_) in MklPoolingOpBase()
1020 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute()
1464 inline memory::format TFDataFormatToMklDnnDataFormat(TensorFormat format) { in TFDataFormatToMklDnnDataFormat() function1519 CHECK_NE(TFDataFormatToMklDnnDataFormat(format), in TFShapeToMklDnnDimsInNCHW()1552 CHECK_NE(TFDataFormatToMklDnnDataFormat(format), in MklDnnDimsInNCHW()