Searched refs:cloned_parameter_node (Results 1 – 2 of 2) sorted by relevance
456 for (auto &cloned_parameter_node : root->parameters()) { in SetClonedTensorShapeForOptimizer() local457 MS_EXCEPTION_IF_NULL(cloned_parameter_node); in SetClonedTensorShapeForOptimizer()458 auto cloned_parameter = cloned_parameter_node->cast<ParameterPtr>(); in SetClonedTensorShapeForOptimizer()461 if (!ParameterIsCloned(cloned_parameter_node)) { in SetClonedTensorShapeForOptimizer()502 std::string param_name = cloned_parameter_node->cast<ParameterPtr>()->name(); in SetClonedTensorShapeForOptimizer()508 MS_EXCEPTION_IF_NULL(cloned_parameter_node->abstract()); in SetClonedTensorShapeForOptimizer()510 auto cloned_abstract = cloned_parameter_node->abstract()->Clone(); in SetClonedTensorShapeForOptimizer()528 cloned_parameter_node->set_abstract(cloned_abstract); in SetClonedTensorShapeForOptimizer()
2697 for (auto &cloned_parameter_node : root->parameters()) { in CheckpointStrategy() local2698 MS_EXCEPTION_IF_NULL(cloned_parameter_node); in CheckpointStrategy()2699 auto cloned_parameter = cloned_parameter_node->cast<ParameterPtr>(); in CheckpointStrategy()2702 if (!ParameterIsCloned(cloned_parameter_node)) { in CheckpointStrategy()2705 std::string cloned_param_name = cloned_parameter_node->cast<ParameterPtr>()->name(); in CheckpointStrategy()2706 auto cloned_param_layout = cloned_parameter_node->user_data<TensorLayout>(); in CheckpointStrategy()