Update CPU and GPU shape in caliper script