Searched refs:param_shapes (Results 1 – 1 of 1) sorted by relevance
30 std::map<std::string, Shape> param_shapes; variable209 param_shapes.clear(); in ParallelParameterContextInitShape()224 param_shapes.clear(); in ParallelParameterContextInitShape()243 auto iter = param_shapes.find(param_node->name()); in ParallelParameterContextRestoreShape()244 if (iter == param_shapes.end()) { in ParallelParameterContextRestoreShape()269 auto ret = param_shapes.try_emplace(param_node->name(), shape); in ParallelParameterContextCkptShape()