add _output_axes_def and _input_axes_def class variables.