Fix open-source dependence on dimension_value.
Fix open-source dependence on dimension_value.

PiperOrigin-RevId: 219223793

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor