Browse Source

Pre Merge pull request !997 from 梁昊/development

pull/997/MERGE
梁昊 Gitee 5 years ago
parent
commit
16a1d61c8b
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ge/graph/preprocess/graph_preprocess.cc

+ 2
- 1
ge/graph/preprocess/graph_preprocess.cc View File

@@ -135,7 +135,8 @@ OpDescPtr CreateTensorShape(const GeTensorDesc &data_tensor) {
void AddTransNodeAttr(const std::string &node_type, const GeTensorDesc &input, const GeTensorDesc &output,
OpDescPtr &op_desc) {
// For format transfer node, the IR definition has src/dst format attrs
if (node_type == TRANSDATA) {

if (node_type == TRANSDATA) {
GE_IF_BOOL_EXEC(
!AttrUtils::SetStr(op_desc, FORMAT_TRANSFER_SRC_FORMAT, TypeUtils::FormatToSerialString(input.GetFormat())),
GELOGW("SetStr FORMAT_TRANSFER_SRC_FORMAT failed");)


Loading…
Cancel
Save