Home
last modified time | relevance | path

Searched refs:my_tfreader_op (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/tests/ut/cpp/dataset/
DtfReader_op_test.cc52 std::shared_ptr<TFReaderOp> my_tfreader_op = in TEST_F() local
55 rc = my_tfreader_op->Init(); in TEST_F()
57 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
60 rc = my_tree->AssignRoot(my_tfreader_op); in TEST_F()
109 std::shared_ptr<TFReaderOp> my_tfreader_op = in TEST_F() local
112 rc = my_tfreader_op->Init(); in TEST_F()
114 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
117 rc = my_tree->AssignRoot(my_tfreader_op); in TEST_F()
166 std::shared_ptr<TFReaderOp> my_tfreader_op = in TEST_F() local
169 rc = my_tfreader_op->Init(); in TEST_F()
[all …]
Dshuffle_op_test.cc61 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
63 rc = my_tfreader_op->Init(); in TEST_F()
65 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
73 rc = my_shuffle_op->AddChild(my_tfreader_op); in TEST_F()
133 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
135 rc = my_tfreader_op->Init(); in TEST_F()
137 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
144 rc = my_shuffle_op->AddChild(my_tfreader_op); in TEST_F()
202 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
204 rc = my_tfreader_op->Init(); in TEST_F()
[all …]
Dzip_op_test.cc62 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
64 rc = my_tfreader_op->Init(); in TEST_F()
66 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
82 rc = zip_op->AddChild(std::move(my_tfreader_op)); in TEST_F()
143 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
145 rc = my_tfreader_op->Init(); in TEST_F()
148 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
150 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
165 my_tfreader_op->SetTotalRepeats(num_repeats); in TEST_F()
166 my_tfreader_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
[all …]
Dmap_op_test.cc118 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in CreateTFReaderOp() local
120 (void)my_tfreader_op->Init(); in CreateTFReaderOp()
121 return my_tfreader_op; in CreateTFReaderOp()
147 auto my_tfreader_op = this->CreateTFReaderOp(); in TEST_F() local
148 rc = my_tree_->AssociateNode(my_tfreader_op); in TEST_F()
161 rc = my_map_op->AddChild(my_tfreader_op); in TEST_F()
197 auto my_tfreader_op = this->CreateTFReaderOp(); in TEST_F() local
198 rc = my_tree_->AssociateNode(my_tfreader_op); in TEST_F()
213 rc = my_map_op->AddChild(my_tfreader_op); in TEST_F()
249 auto my_tfreader_op = this->CreateTFReaderOp(); in TEST_F() local
[all …]
Dclient_config_test.cc95 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
97 rc = my_tfreader_op->Init(); in TEST_F()
99 ASSERT_EQ(my_tfreader_op->NumWorkers(), 1); in TEST_F()
100 my_tree->AssociateNode(my_tfreader_op); in TEST_F()
103 my_tree->AssignRoot(my_tfreader_op); in TEST_F()
121 ASSERT_EQ(my_tfreader_op->NumWorkers(), 1); in TEST_F()
Dtake_op_test.cc52 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
54 rc = my_tfreader_op->Init(); in TEST_F()
59 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
65 rc = my_take_op->AddChild(my_tfreader_op); in TEST_F()
Dskip_op_test.cc45 std::shared_ptr<TFReaderOp> my_tfreader_op = in TEST_F() local
48 rc = my_tfreader_op->Init(); in TEST_F()
50 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
59 rc = skip_op->AddChild(my_tfreader_op); in TEST_F()
Dproject_op_test.cc50 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
52 rc = my_tfreader_op->Init(); in TEST_F()
54 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
64 rc = my_project_op->AddChild(my_tfreader_op); in TEST_F()
Drename_op_test.cc60 std::shared_ptr<TFReaderOp> my_tfreader_op = in TEST_F() local
63 rc = my_tfreader_op->Init(); in TEST_F()
65 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
76 rc = rename_op->AddChild(std::move(my_tfreader_op)); in TEST_F()
Dexecution_tree_test.cc105 std::shared_ptr<TFReaderOp> my_tfreader_op = std::make_shared<TFReaderOp>( in TEST_F() local
107 rc = my_tfreader_op->Init(); in TEST_F()
109 rc = my_tree->AssociateNode(my_tfreader_op); in TEST_F()
111 rc = my_tree->AssignRoot(my_tfreader_op); in TEST_F()