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