Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dtpu_test.cc36 EXPECT_FALSE(IsTPUGraphDef(tpu_graph)); in TEST_F()
45 EXPECT_TRUE(IsTPUGraphDef(tpu_graph)); in TEST_F()
57 EXPECT_TRUE(IsTPUGraphDef(tpu_graph)); in TEST_F()
Dtpu.h25 bool IsTPUGraphDef(const GraphDef& def);
Dtpu.cc24 bool IsTPUGraphDef(const GraphDef& def) { in IsTPUGraphDef() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer.cc304 if (IsTPUGraphDef(*optimized_graph)) { in Optimize()
Dmeta_optimizer.cc852 if (IsTPUGraphDef(*optimized_graph)) { in OptimizeConsumeItem()