Searched refs:tensorflow_op_name (Results 1 – 4 of 4) sorted by relevance
22 bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) { in IsWhitelistedFlexOp() argument469 return whitelisted_flex_ops->find(tensorflow_op_name) != in IsWhitelistedFlexOp()
30 bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name);
119 const string& tensorflow_op_name);
2592 const string& tensorflow_op_name) { in ShouldExportAsFlexOp() argument2602 ->LookUpOpDef(tensorflow_op_name, &op_def) in ShouldExportAsFlexOp()2607 if (!IsWhitelistedFlexOp(tensorflow_op_name)) { in ShouldExportAsFlexOp()2608 LOG(WARNING) << "Op " << tensorflow_op_name in ShouldExportAsFlexOp()