Fix extract_features

Removed unnecessary "mutable_cpu_data".