Home
last modified time | relevance | path

Searched refs:columns_to_project (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/tests/ut/cpp/dataset/
Dproject_op_test.cc58 std::vector<std::string> columns_to_project = {"col_sint16", "col_float", "col_2d"}; in TEST_F() local
59 std::shared_ptr<ProjectOp> my_project_op = std::make_shared<ProjectOp>(columns_to_project); in TEST_F()
87 ASSERT_EQ(tensor_list.size(), columns_to_project.size()); in TEST_F()
Dc_api_pull_based_test.cc70 std::vector<std::string> columns_to_project = {"image"}; in TEST_F() local
71 ds2 = ds2->Project(columns_to_project); in TEST_F()
Dmap_op_test.cc263 std::vector<std::string> columns_to_project = {"X", "Y", "Z", "label", "A", "B"}; in TEST_F() local
264 std::shared_ptr<ProjectOp> my_project_op = std::make_shared<ProjectOp>(columns_to_project); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dproject_op.cc29 ProjectOp::ProjectOp(const std::vector<std::string> &columns_to_project) in ProjectOp() argument
30 : PipelineOp(0), columns_to_project_(columns_to_project) {} in ProjectOp()
Dproject_op.h31 explicit ProjectOp(const std::vector<std::string> &columns_to_project);