class_docstr: ReshapeGenerator is the base class for all the reshape operation.
class_docstr_tok: ["ReshapeGenerator","is","the","base","class","for","all","the","reshape","operation","."]
code: def update_compute_cost(self, strategy: ShardingStrategy):
      compute_cost = TrainCycleItem(fwd=10, bwd=10, total=20)
      strategy.compute_cost = compute_cost
  
imports: import copy
  from typing import List
  from colossalai.auto_parallel.tensor_shard.node_handler.strategy.strategy_generator import FollowingStrategyGenerator
  from colossalai.auto_parallel.tensor_shard.sharding_strategy import CommAction, CommType, MemoryCost, ShardingStrategy, TrainCycleItem
  from colossalai.auto_parallel.tensor_shard.utils import check_keep_sharding_status, detect_reshape_mapping, infer_output_dim_partition_dict
  from colossalai.tensor.shape_consistency import CollectiveCommPattern
  from colossalai.tensor.sharding_spec import ShardingSpec
label_desc: Use image node_id 2 for calling the ReshapeGenerator obj's underlying member method code with example usage: obj.update_compute_cost(strategy) without return types