Searched refs:TFRecordNode (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/ |
| D | tf_record_node.cc | 35 std::shared_ptr<DatasetNode> TFRecordNode::Copy() { in Copy()
36 std::shared_ptr<TFRecordNode> node; in Copy()
38 …node = std::make_shared<TFRecordNode>(dataset_files_, schema_obj_, columns_list_, num_samples_, sh… in Copy()
41 …node = std::make_shared<TFRecordNode>(dataset_files_, schema_path_, columns_list_, num_samples_, s… in Copy()
47 void TFRecordNode::Print(std::ostream &out) const { in Print()
53 Status TFRecordNode::ValidateParams() { in ValidateParams()
113 Status TFRecordNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
164 Status TFRecordNode::GetShardId(int32_t *const shard_id) { in GetShardId()
171 Status TFRecordNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool est… in GetDatasetSize()
198 Status TFRecordNode::GetShardFileList(std::vector<std::string> *shard_filenames) { in GetShardFileList()
[all …]
|
| D | tf_record_node.h | 32 class TFRecordNode : public NonMappableSourceNode {
38 TFRecordNode(const std::vector<std::string> &dataset_files, std::string schema, in TFRecordNode() function
59 TFRecordNode(const std::vector<std::string> &dataset_files, std::shared_ptr<SchemaObj> schema, in TFRecordNode() function
73 ~TFRecordNode() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/ |
| D | bindings.cc | 292 PYBIND_REGISTER(TFRecordNode, 2, ([](const py::module *m) { in __anon810239692602()
293 … (void)py::class_<TFRecordNode, DatasetNode, std::shared_ptr<TFRecordNode>>(*m, "TFRecordNode", in __anon810239692602()
298 std::shared_ptr<TFRecordNode> tfrecord = std::make_shared<TFRecordNode>( in __anon810239692602()
307 std::shared_ptr<TFRecordNode> tfrecord = std::make_shared<TFRecordNode>( in __anon810239692602()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/ |
| D | pass.h | 49 class TFRecordNode; variable
77 class TFRecordNode; variable
213 virtual Status Visit(std::shared_ptr<TFRecordNode> node, bool *const modified);
214 virtual Status VisitAfter(std::shared_ptr<TFRecordNode> node, bool *const modified);
|
| D | pass.cc | 252 Status IRNodePass::Visit(std::shared_ptr<TFRecordNode> node, bool *const modified) { in Visit()
255 Status IRNodePass::VisitAfter(std::shared_ptr<TFRecordNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | cache_validation_pass.h | 96 Status VisitAfter(std::shared_ptr<TFRecordNode> node, bool *const modified) override;
|
| D | cache_validation_pass.cc | 161 Status CacheValidationPass::VisitAfter(std::shared_ptr<TFRecordNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | serdes.cc | 174 RETURN_IF_NOT_OK(TFRecordNode::from_json(json_obj, ds)); in CreateDatasetNode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 1336 auto ds = std::make_shared<TFRecordNode>(VectorCharToString(dataset_files), CharToString(schema), in TFRecordDataset()
1345 …auto ds = std::make_shared<TFRecordNode>(VectorCharToString(dataset_files), schema, VectorCharToSt… in TFRecordDataset()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | deserialize_test.cc | 413 …std::make_shared<TFRecordNode>(dataset_files, schema, columns_list, num_samples, ShuffleMode::kFil… in TEST_F()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 4313 …return cde.TFRecordNode(self.dataset_files, schema, self.columns_list, self.num_samples, self.shuf…
|