infershape paralelly

pull/970/head
wxl 4 years ago
parent 7bb0eeb614
commit 4d1f43053c

@ -68,7 +68,6 @@ Status ShapeInferenceEngine::InferShape(NodeState &node_state) {
// Do shape inference
GELOGD("[%s] Start to invoke InferShapeAndType", node_item.NodeName().c_str());
{
std::lock_guard<std::mutex> lk(mu_);
RECORD_SHAPE_INFERENCE_EVENT(execution_context_, node_item.NodeName().c_str(), "[InferShapeAndType] Start");
GE_CHK_STATUS_RET(ShapeRefiner::InferShapeAndTypeForRunning(node_item.node, true),
"Invoke InferShapeAndType failed.");

Loading…
Cancel
Save