public GraphDef ToGraphDef() { TF_Buffer b = this.ToGraphDefBuffer(); var stream = CodedInputStream.CreateWithLimits((UnmanagedMemoryStream)b, 256 * 1024 * 1024, 100); var def = GraphDef.Parser.ParseFrom(stream); b.Delete(); return(def); }
private GraphDef _as_graph_def(bool add_shapes = false) { GraphDef def; using (var status = new Status()) using (var buffer = ToGraphDef(status)) { status.Check(true); // limit size to 250M, recursion to max 100 var inputStream = CodedInputStream.CreateWithLimits(buffer.DangerousMemoryBlock, 250 * 1024 * 1024, 100); def = GraphDef.Parser.ParseFrom(inputStream); } // Strip the experimental library field iff it's empty. // if(def.Library.Function.Count == 0) return(def); }
public static GraphDef Import(string filePath) => GraphDef.Parser.ParseFrom(CodedInputStream.CreateWithLimits(File.OpenRead(filePath), 256 * 1024 * 1024, 100));