Searched refs:index_fields (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/mindrecord/ |
| D | filewriter.py | 179 def add_index(self, index_fields): argument
202 if not index_fields or not isinstance(index_fields, list):
205 for field in index_fields:
210 return self._header.add_index_fields(index_fields)
|
| D | shardheader.py | 56 def add_index_fields(self, index_fields): argument
69 ret = self._header.add_index_fields(index_fields)
|
| /third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/ |
| D | shard_header.h | 139 … const std::vector<std::string> &index_fields, std::vector<std::string> &blob_fields,
160 Status ParseIndexFields(const json &index_fields);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/meta/ |
| D | shard_header.cc | 214 Status ShardHeader::ParseIndexFields(const json &index_fields) { in ParseIndexFields() argument
216 for (auto &index_field : index_fields) { in ParseIndexFields()
632 … const std::vector<std::string> &index_fields, std::vector<std::string> &blob_fields, in Initialize() argument
640 if (!index_fields.empty()) { in Initialize()
641 (void)transform(index_fields.begin(), index_fields.end(), std::back_inserter(id_index_fields), in Initialize()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/ |
| D | tree_consumer.cc | 284 std::vector<std::string> index_fields; in Save() local
285 RETURN_IF_NOT_OK(FetchMetaFromTensorRow(column_name_id_map, row, &mr_json, &index_fields)); in Save()
288 …mindrecord::ShardHeader::Initialize(&mr_header, mr_json, index_fields, blob_fields, mr_schema_id)); in Save()
357 std::vector<std::string> *index_fields) { in FetchMetaFromTensorRow() argument
361 if (index_fields == nullptr) { in FetchMetaFromTensorRow()
404 index_fields->emplace_back(column_name); // candidate of index fields in FetchMetaFromTensorRow()
|
| D | tree_consumer.h | 129 … const TensorRow &row, nlohmann::json *schema, std::vector<std::string> *index_fields);
|
| /third_party/mindspore/tests/ut/python/mindrecord/ |
| D | skip_test_issue.py | 156 index_fields = []
157 ret = header.add_index_fields(index_fields)
|
| /third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/io/ |
| D | shard_index_generator.cc | 475 std::vector<std::pair<uint64_t, std::string>> index_fields = shard_header_.GetFields(); in GenerateIndexFields() local
476 for (const auto &field : index_fields) { in GenerateIndexFields()
|
| D | shard_reader.cc | 890 auto index_fields = shard_header_->GetFields(); in GetNumClasses() local
893 for (auto &field : index_fields) { in GetNumClasses()
|