|
@@ -68,6 +68,7 @@ Status ShapeInferenceEngine::InferShape(NodeState &node_state) { |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Do shape inference |
|
|
// Do shape inference |
|
|
|
|
|
// Skipping infer shape of input node. |
|
|
GELOGD("[%s] Start to invoke InferShapeAndType", node_item.NodeName().c_str()); |
|
|
GELOGD("[%s] Start to invoke InferShapeAndType", node_item.NodeName().c_str()); |
|
|
if (node_state.GetType() != DATA_TYPE && node_state.GetType() != AIPP_DATA_TYPE) { |
|
|
if (node_state.GetType() != DATA_TYPE && node_state.GetType() != AIPP_DATA_TYPE) { |
|
|
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"); |
|
|