Browse Source

Fix ut.

tags/v1.3.0
zhaozhixuan 4 years ago
parent
commit
24eedfa3b4
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/hybrid/executor/worker/shape_inference_engine.cc

+ 1
- 0
ge/hybrid/executor/worker/shape_inference_engine.cc View File

@@ -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");


Loading…
Cancel
Save