|
|
|
@@ -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}; |
|
|
|
|