Lines Matching defs:op_def
179 const OpDef* op_def;
180 TF_RETURN_IF_ERROR(lib_def->LookUpOpDef(node_def.op(), &op_def));
185 if (IsOpWhitelisted(op_def)) {
188 if (op_def->is_stateful()) {
201 // Also looks up the `op_def->name` in the global
203 bool IsOpWhitelisted(const OpDef* op_def) const {
204 return (StringPiece(op_def->name()).ends_with("Dataset") &&
205 op_def->output_arg_size() == 1 &&
206 op_def->output_arg(0).type() == DT_VARIANT) ||
208 op_def->name());
213 bool HasAttr(const OpDef* op_def, const string& attr_name) const {
214 for (auto attr : op_def->attr()) {
