Searched defs:restore_op (Results 1 – 7 of 7) sorted by relevance
73 ImmediateOpPtr restore_op(ctx->CreateOperation()); in SingleRestore() local
56 if (auto restore_op = llvm::dyn_cast_or_null<mlir::TF::RestoreV2Op>(op)) { in BackwardShapeInferenceToRestoreOp() local
387 for (const RestoreOp& restore_op : restore_ops) { in RestoreTensorsV2() local438 for (const RestoreOp& restore_op : restore_ops) { in RestoreTensorsV2() local
62 string restore_op; member
793 mlir::ModuleOp module, mlir::TF::RestoreV2Op* restore_op) { in GetLayoutsFromAssignVariableOps()
292 Node* restore_op; in AddRestoreVariableSubgraphs() local
181 def restore_op(self, filename_tensor, saveable, preferred_shard): member in BaseSaverBuilder