Tensorflow-MKL выдает ошибки в 3D-данных

Я обучаю модель для выполнения объемной сегментации (3D-данные). Я тренируюсь на CPU из-за размера входных данных. Я использую среду anaconda с tenorflow-mkl и keras. Когда я тренирую модель, у меня получается error:

"tensorflow.python.framework.errors_impl.InvalidArgumentError: Value for attr 'data_format' of "NDHWC" is not in the list of allowed values: "NHWC", "NCHW"

Однако на их github говорится, что он работает с объемной сегментацией (https://github.com/intel/mkl-dnn). Как я могу решить эту проблему, чтобы обучить свой 3D-Unet с mkl?