Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmeta_optimizer.cc122 if (!data::IsTFDataFunction(*func)) continue; in Optimize()
Dfusion_utils.cc435 if (data::IsTFDataFunction(func)) known_attribute_size += 1; in FuseFunctions()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc662 if (data::IsTFDataFunction(func_def)) in PropagateTFDataAttrs()
671 !data::IsTFDataFunction(func_def)) { in PropagateTFDataAttrs()
814 if (data::IsTFDataFunction(func)) continue; in OptimizeConsumeItem()
Dmeta_optimizer_test.cc1211 EXPECT_EQ(data::IsTFDataFunction(*outer_func_after_opt), in TEST_P()
1214 EXPECT_EQ(data::IsTFDataFunction(*inner_func_after_opt), true); in TEST_P()
1216 EXPECT_EQ(data::IsTFDataFunction(*inner_func_after_opt), false); in TEST_P()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h78 inline bool IsTFDataFunction(const FunctionDef& func) { in IsTFDataFunction() function