Home
last modified time | relevance | path

Searched refs:ShardColumn (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/meta/
Dshard_column.cc25 ShardColumn::ShardColumn(const std::shared_ptr<ShardHeader> &shard_header, bool compress_integer) { in ShardColumn() function in mindspore::mindrecord::ShardColumn
31 ShardColumn::ShardColumn(const json &schema_json, bool compress_integer) { Init(schema_json, compre… in ShardColumn() function in mindspore::mindrecord::ShardColumn
33 void ShardColumn::Init(const json &schema_json, bool compress_integer) { in Init()
75 Status ShardColumn::GetColumnTypeByName(const std::string &column_name, ColumnDataType *column_data… in GetColumnTypeByName()
94 Status ShardColumn::GetColumnValueByName(const std::string &column_name, const std::vector<uint8_t>… in GetColumnValueByName()
126 Status ShardColumn::GetColumnFromJson(const std::string &column_name, const json &columns_json, in GetColumnFromJson()
173 Status ShardColumn::GetFloat(std::unique_ptr<unsigned char[]> *data_ptr, const json &json_column_va… in GetFloat()
202 Status ShardColumn::GetInt(std::unique_ptr<unsigned char[]> *data_ptr, const json &json_column_valu… in GetInt()
246 Status ShardColumn::GetColumnFromBlob(const std::string &column_name, const std::vector<uint8_t> &c… in GetColumnFromBlob()
265 ColumnCategory ShardColumn::CheckColumnName(const std::string &column_name) { in CheckColumnName()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/
Dgraph_feature_parser.h38 using mindrecord::ShardColumn;
42 explicit GraphFeatureParser(const ShardColumn &shard_column);
62 std::unique_ptr<ShardColumn> shard_column_;
Dgraph_feature_parser.cc30 GraphFeatureParser::GraphFeatureParser(const ShardColumn &shard_column) { in GraphFeatureParser()
31 shard_column_ = std::make_unique<ShardColumn>(shard_column); in GraphFeatureParser()
Dgraph_data_client.cc603 graph_feature_parser_ = std::make_unique<GraphFeatureParser>(ShardColumn(data_schema_)); in InitFeatureParser()
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/
Dshard_column.h60 class __attribute__((visibility("default"))) ShardColumn {
62 …explicit ShardColumn(const std::shared_ptr<ShardHeader> &shard_header, bool compress_integer = tru…
63 explicit ShardColumn(const json &schema_json, bool compress_integer = true);
65 ~ShardColumn() = default;
Dshard_reader.h110 std::shared_ptr<ShardColumn> GetShardColumn() const;
300 std::shared_ptr<ShardColumn> shard_column_; // shard column
Dshard_writer.h244 std::shared_ptr<ShardColumn> shard_column_; // shard columns
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/io/
Dshard_reader.cc101 shard_column_ = std::make_shared<ShardColumn>(shard_header_, false); in Init()
103 shard_column_ = std::make_shared<ShardColumn>(shard_header_, true); in Init()
287 std::shared_ptr<ShardColumn> ShardReader::GetShardColumn() const { return shard_column_; } in GetShardColumn()
Dshard_writer.cc162 shard_column_ = std::make_shared<ShardColumn>(shard_header_); in OpenForAppend()
209 shard_column_ = std::make_shared<ShardColumn>(shard_header_); in SetShardHeader()