xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Alarm modification

This commit is contained in:
shenwei41 2020-09-08 11:00:10 +08:00
parent 0d7bca087b
commit 0c6ce88058
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/minddata/dataset/engine/datasetops/source/mindrecord_op.cc
View File

@ -52,6 +52,7 @@ MindRecordOp::Builder::Builder() : build_dataset_file_({}) {
build_rows_per_buffer_ = cfg->rows_per_buffer();
build_op_connector_queue_size_ = cfg->op_connector_size();
builder_num_workers_ = 0;
build_load_dataset_ = false;
build_num_padded_ = 0;
build_sample_ = nullptr;
}
@ -344,6 +345,7 @@ Status MindRecordOp::LoadTensorRow(TensorRow *tensor_row, const std::vector<uint
DataType type = column.type();
// Set shape
CHECK_FAIL_RETURN_UNEXPECTED(column_data_type_size != 0, "The divisor cannot be 0.");
auto num_elements = n_bytes / column_data_type_size;
if (type == DataType::DE_STRING) {
std::string s{data, data + n_bytes};