refactor_for_functional_conv