|
|
|
@ -102,9 +102,8 @@ void RandomDataOp::Print(std::ostream &out, bool show_all) const {
|
|
|
|
|
// Call the super class for displaying any common detailed info
|
|
|
|
|
ParallelOp::Print(out, show_all);
|
|
|
|
|
// Then show any custom derived-internal stuff
|
|
|
|
|
out << "\nTotal_rows: " << total_rows_
|
|
|
|
|
<< "\nRows per buffer: " << rows_per_buffer_
|
|
|
|
|
<< "\nSchema:\n" << *data_schema_ << "\n\n";
|
|
|
|
|
out << "\nTotal_rows: " << total_rows_ << "\nRows per buffer: " << rows_per_buffer_ << "\nSchema:\n"
|
|
|
|
|
<< *data_schema_ << "\n\n";
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -143,8 +142,7 @@ Status RandomDataOp::GenerateSchema() {
|
|
|
|
|
|
|
|
|
|
// Create the column descriptor
|
|
|
|
|
std::string colName = "c" + std::to_string(i);
|
|
|
|
|
newCol = std::make_unique<ColDescriptor>(colName, DataType(newType), TensorImpl::kFlexible, rank,
|
|
|
|
|
newShape.get());
|
|
|
|
|
newCol = std::make_unique<ColDescriptor>(colName, DataType(newType), TensorImpl::kFlexible, rank, newShape.get());
|
|
|
|
|
|
|
|
|
|
data_schema_->AddColumn(*newCol);
|
|
|
|
|
}
|
|
|
|
@ -411,4 +409,3 @@ Status RandomDataOp::Reset() {
|
|
|
|
|
}
|
|
|
|
|
} // namespace dataset
|
|
|
|
|
} // namespace mindspore
|
|
|
|
|
|
|
|
|
|