Searched refs:FORMAT_NDHWC (Results 1 – 3 of 3) sorted by relevance
96 #define MKL_TENSOR_FORMAT_NDHWC MklTensorFormat::FORMAT_NDHWC
164 FORMAT_NDHWC = 2, enumerator202 } else if (format == MklTensorFormat::FORMAT_NDHWC) {1012 if (format == MklTensorFormat::FORMAT_NDHWC) return memory::format_tag::ndhwc; in MklTensorFormatToMklDnnDataFormat()1025 if (format == FORMAT_NHWC) return MklTensorFormat::FORMAT_NDHWC; in TFDataFormatToMklDnn3DDataFormat()1050 format == MklTensorFormat::FORMAT_NDHWC) in MklDnnDataFormatToTFDataFormat()
253 layout_type = MklTensorFormat::FORMAT_NDHWC; in Compute()