Searched defs:main_op (Results 1 – 3 of 3) sorted by relevance
34 def main_op(): function
439 def _add_collections(self, assets_collection, main_op, train_op): argument467 def _maybe_add_main_op(self, main_op): argument522 main_op=None, argument567 main_op=None, argument
1575 FuncOp main_op, const std::string& saved_model_tag) { in BuildSignaturedef()