Searched refs:tf_data_functions (Results 1 – 1 of 1) sorted by relevance
986 absl::flat_hash_set<std::string> tf_data_functions; in PropagateTFDataAttrs() local994 if (tf_data_functions.contains(func_name)) return; in PropagateTFDataAttrs()998 tf_data_functions.insert(func_name); in PropagateTFDataAttrs()1031 if (tf_data_functions.contains(func_name) && in PropagateTFDataAttrs()