Browse Source

fix sc

pull/236/head
wjm 5 years ago
parent
commit
81bd0f0f5f
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      parser/tensorflow/tensorflow_parser.cc

+ 2
- 1
parser/tensorflow/tensorflow_parser.cc View File

@@ -2872,7 +2872,8 @@ Status TensorFlowModelParser::GetFormatTranspose(const NodeDef *transpose_node,
return FAILED);
const TensorProto &tensor = attr_value.tensor();
const TensorShapeProto &tensor_shape = tensor.tensor_shape();
GE_IF_BOOL_EXEC(tensor_shape.dim_size() != 1 || tensor_shape.dim(0).size() != parser::DIM_DEFAULT_SIZE, return SUCCESS);
GE_IF_BOOL_EXEC(tensor_shape.dim_size() != 1 || tensor_shape.dim(0).size() != parser::DIM_DEFAULT_SIZE,
return SUCCESS);
GE_IF_BOOL_EXEC(tensor.tensor_content().empty(), return SUCCESS);

vector<int64_t> perm_value;


Loading…
Cancel
Save