Searched refs:tree_adapter (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | ir_tree_adapter_test.cc | 53 mindspore::dataset::TreeAdapter tree_adapter; in TEST_F() local
55 Status rc = tree_adapter.Compile(ds->IRNode(), 1); in TEST_F()
60 EXPECT_EQ(tree_adapter.GetColumnNameMap(), map); in TEST_F()
66 rc = tree_adapter.GetNext(&row); in TEST_F()
71 rc = tree_adapter.GetNext(&row); in TEST_F()
88 mindspore::dataset::TreeAdapter tree_adapter; in TEST_F() local
90 Status rc = tree_adapter.Compile(ds->IRNode(), 2); in TEST_F()
93 const std::unordered_map<std::string, int32_t> map = tree_adapter.GetColumnNameMap(); in TEST_F()
94 EXPECT_EQ(tree_adapter.GetColumnNameMap(), map); in TEST_F()
100 rc = tree_adapter.GetNext(&row); in TEST_F()
[all …]
|
| D | ir_callback_test.cc | 343 TreeAdapter tree_adapter; in TEST_F() local
345 ASSERT_OK(tree_adapter.Compile(ds->IRNode(), 1)); in TEST_F()
348 ASSERT_OK(tree_adapter.GetNext(&row)); in TEST_F()
350 ASSERT_OK(tree_adapter.GetNext(&row)); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/ |
| D | python_tree_consumer.cc | 66 Status PythonDatasetSizeGetter::GetRow(const std::shared_ptr<TreeAdapter> &tree_adapter, TensorRow … in GetRow() argument
67 RETURN_UNEXPECTED_IF_NULL(tree_adapter); in GetRow()
70 return DatasetSizeGetter::GetRow(tree_adapter, r); in GetRow()
|
| D | pull_based_tree_consumer.cc | 72 std::unique_ptr<TreeAdapter> tree_adapter = std::make_unique<TreeAdapter>(); in GetNextAsOrderedPair() local
84 for (const auto &itr : tree_adapter->GetColumnNameMap()) { in GetNextAsOrderedPair()
|
| D | tree_consumer.cc | 717 …std::shared_ptr<TreeAdapter> tree_adapter = std::make_shared<TreeAdapter>(TreeAdapter::UsageFlag::… in DryRun() local
718 tree_adapters_.push_back(tree_adapter); in DryRun()
719 RETURN_IF_NOT_OK(tree_adapter->Compile(ir_node, 1)); in DryRun()
721 RETURN_IF_NOT_OK(GetRow(tree_adapter, &row)); in DryRun()
725 RETURN_IF_NOT_OK(GetRow(tree_adapter, &row)); in DryRun()
730 Status DatasetSizeGetter::GetRow(const std::shared_ptr<TreeAdapter> &tree_adapter, TensorRow *row) { in GetRow() argument
732 return tree_adapter->GetNext(row); in GetRow()
|
| D | python_tree_consumer.h | 67 Status GetRow(const std::shared_ptr<TreeAdapter> &tree_adapter, TensorRow *r) override;
|
| D | tree_consumer.h | 235 virtual Status GetRow(const std::shared_ptr<TreeAdapter> &tree_adapter, TensorRow *row);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | CMakeLists.txt | 23 tree_adapter.cc
|
| /third_party/mindspore/mindspore/lite/minddata/ |
| D | CMakeLists.txt | 195 ${MINDDATA_DIR}/engine/tree_adapter.cc
|
| /third_party/mindspore/mindspore/lite/tools/dataset/cropper/ |
| D | dependencies.txt | 1 …tree_adapter.cc", "mindspore/lite/build/_deps/jpeg_turbo-src/./jmorecfg.h", "mindspore/ccsrc/mindd…
|