Home
last modified time | relevance | path

Searched refs:PruneGraphDefInto (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util_test.cc168 TEST(PruneGraphDefInto, Basic) { in TEST() argument
176 ExpectErrorContains(PruneGraphDefInto(FetchesConfig({"missing"}), def, &copy), in TEST()
178 ExpectErrorContains(PruneGraphDefInto(FetchesConfig({"a"}), def, &copy), in TEST()
183 ExpectErrorContains(PruneGraphDefInto(FetchesConfig({"a"}), def, &copy), in TEST()
198 TF_EXPECT_OK(PruneGraphDefInto(FetchesConfig({"a"}), def, &copy)); in TEST()
208 TF_EXPECT_OK(PruneGraphDefInto(FetchesConfig({"a"}), def, &copy)); in TEST()
213 TF_EXPECT_OK(PruneGraphDefInto(FetchesConfig({"a", "e"}), def, &copy)); in TEST()
Dtf2xla_util.h46 Status PruneGraphDefInto(const tf2xla::Config& config, const GraphDef& in,
Dtf2xla.cc360 PruneGraphDefInto(config, first_copy_def, &second_copy_def)); in InitGraph()
Dtf2xla_util.cc431 Status PruneGraphDefInto(const tf2xla::Config& config, const GraphDef& in, in PruneGraphDefInto() function