Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_input_ops.cc379 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
Dmkl_conv_grad_filter_ops.cc440 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
Dmkl_pooling_ops_common.h475 : TFDataFormatToMklDnn3DDataFormat(this->data_format_tf_); in MklPoolingOpBase()
Dmkl_conv_ops.cc1021 : TFDataFormatToMklDnn3DDataFormat(data_format_); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1450 inline memory::format TFDataFormatToMklDnn3DDataFormat(TensorFormat format) { in TFDataFormatToMklDnn3DDataFormat() function
1534 CHECK_NE(TFDataFormatToMklDnn3DDataFormat(format), in TFShapeToMklDnnDimsInNCDHW()