|
|
@@ -62,21 +62,7 @@ Status ShapeInferenceEngine::InferShape(NodeState &node_state) { |
|
|
|
{ |
|
|
|
std::lock_guard<std::mutex> lk(mu_); |
|
|
|
RECORD_SHAPE_INFERENCE_EVENT(execution_context_, node_item.NodeName().c_str(), "[InferShapeAndType] Start"); |
|
|
|
<<<<<<< Updated upstream |
|
|
|
======= |
|
|
|
<<<<<<< Updated upstream |
|
|
|
>>>>>>> Stashed changes |
|
|
|
<<<<<<< HEAD |
|
|
|
GE_CHK_STATUS_RET(ShapeRefiner::InferShapeAndTypeForRunning(node_item.node, true), "Invoke InferShapeAndType failed."); |
|
|
|
======= |
|
|
|
GE_CHK_STATUS_RET(ShapeRefiner::InferShapeAndTypeForRunning(node_item.node, true), "Invoke InferShapeAndType failed."); |
|
|
|
>>>>>>> efe1779b76b2f4508a8eaa330cd169d19c00813d |
|
|
|
<<<<<<< Updated upstream |
|
|
|
======= |
|
|
|
======= |
|
|
|
GE_CHK_STATUS_RET(ShapeRefiner::InferShapeAndTypeForRunning(node_item.node, true), "Invoke InferShapeAndType failed."); |
|
|
|
>>>>>>> Stashed changes |
|
|
|
>>>>>>> Stashed changes |
|
|
|
RECORD_SHAPE_INFERENCE_EVENT(execution_context_, node_item.NodeName().c_str(), "[InferShapeAndType] End"); |
|
|
|
} |
|
|
|
// Check again to make sure shape is valid after shape inference |
|
|
|