Browse Source

Transdata

pull/1211/head
zk 5 years ago
parent
commit
0c993dd696
1 changed files with 9 additions and 7 deletions
  1. +9
    -7
      ge/common/formats/format_transfers/format_transfer_fractal_z.cc

+ 9
- 7
ge/common/formats/format_transfers/format_transfer_fractal_z.cc View File

@@ -494,6 +494,15 @@ Status FormatTransferFractalZ::TransFormat(const TransArgs &args, TransResult &r
}
return TransFormatHwcnToFzWithGroups(args, result, groups);
}

Status FormatTransferFractalZ::TransShape(Format src_format, const std::vector<int64_t> &src_shape, DataType data_type,
Format dst_format, std::vector<int64_t> &dst_shape, int64_t groups){
if (CheckDataTypeSupport(data_type) != SUCCESS) {
return ACL_ERROR_GE_TRANSSHAPE_DATATYPE_INVALID;
}
return TransShapeHwcnToFzWithGroups(src_shape, data_type, dst_shape, groups);
}

Status FormatTransferFractalZ::TransFormat(const TransArgs &args, TransResult &result) {
GELOGD("Begin to trans format from %s to %s, src shape %s, data type %s, dst shape %s",
TypeUtils::FormatToSerialString(args.src_format).c_str(),
@@ -523,13 +532,6 @@ Status FormatTransferFractalZ::TransFormat(const TransArgs &args, TransResult &r
return UNSUPPORTED;
}

Status FormatTransferFractalZ::TransShape(Format src_format, const std::vector<int64_t> &src_shape, DataType data_type,
Format dst_format, std::vector<int64_t> &dst_shape,int64_t groups){
if (CheckDataTypeSupport(data_type) != SUCCESS) {
return ACL_ERROR_GE_TRANSSHAPE_DATATYPE_INVALID;
}
return TransShapeHwcnToFzWithGroups(src_shape, data_type, dst_shape, groups);
}
Status FormatTransferFractalZ::TransShape(Format src_format, const std::vector<int64_t> &src_shape, DataType data_type,
Format dst_format, std::vector<int64_t> &dst_shape) {
if (CheckDataTypeSupport(data_type) != SUCCESS) {


Loading…
Cancel
Save