pull/797/head
wjm 4 years ago
parent a9b4cf400a
commit ba745a12d3

@ -495,8 +495,8 @@ Status MultiBatchClonePass::SetMaxShapeToData(const NodePtr &data) {
int64_t tensor_size = 0; int64_t tensor_size = 0;
(void)TensorUtils::GetTensorSizeInBytes(tensor, tensor_size); (void)TensorUtils::GetTensorSizeInBytes(tensor, tensor_size);
string input_str = TypeUtils::FormatToSerialString(tensor.GetFormat()) + ":" + string input_str = TypeUtils::FormatToSerialString(tensor.GetFormat()) + ":" +
TypeUtils::DataTypeToSerialString(tensor.GetDataType()) + ":" + data->GetName() + ":" + TypeUtils::DataTypeToSerialString(tensor.GetDataType()) + ":" + data->GetName() + ":" +
std::to_string(tensor_size) + ":" + std::to_string(tensor.GetShape().GetDimNum()) + ":" + std::to_string(tensor_size) + ":" + std::to_string(tensor.GetShape().GetDimNum()) + ":" +
formats::JoinToString(tensor.GetShape().GetDims()); formats::JoinToString(tensor.GetShape().GetDims());
input_dims_str.emplace_back(input_str); input_dims_str.emplace_back(input_str);
} }

Loading…
Cancel
Save