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