diff --git a/tensorflow/tools/dockerfiles/spec.yml b/tensorflow/tools/dockerfiles/spec.yml index ec20f63f3d8..6d70faaaf03 100644 --- a/tensorflow/tools/dockerfiles/spec.yml +++ b/tensorflow/tools/dockerfiles/spec.yml @@ -38,6 +38,8 @@ releases: versioned: tag_specs: - "{_TAG_PREFIX}{ubuntu}{jupyter}" + onednn: + tag_specs: - "{_TAG_PREFIX}{ubuntu-1604-onednn}" - "{_TAG_PREFIX}{ubuntu-1804-onednn}" - "{_TAG_PREFIX}{ubuntu-2004-onednn}" @@ -191,6 +193,7 @@ slice_sets: - CHECKOUT_TF_SRC=1 - CHECKOUT_HOROVOD_SRC=1 + ubuntu-1604-onednn: - add_to_name: "-16.04-onednn" dockerfile_exclusive_name: "ubuntu-16.04-onednn"