(t *testing.T)
| 53 | } |
| 54 | |
| 55 | func TestGraphWriteToAndImport(t *testing.T) { |
| 56 | // Construct a graph |
| 57 | g := NewGraph() |
| 58 | v, err := NewTensor(int64(1)) |
| 59 | if err != nil { |
| 60 | t.Fatal(err) |
| 61 | } |
| 62 | input, err := Placeholder(g, "input", v.DataType()) |
| 63 | if err != nil { |
| 64 | t.Fatal(err) |
| 65 | } |
| 66 | if _, err := Neg(g, "neg", input); err != nil { |
| 67 | t.Fatal(err) |
| 68 | } |
| 69 | |
| 70 | // Serialize the graph |
| 71 | buf := new(bytes.Buffer) |
| 72 | if _, err := g.WriteTo(buf); err != nil { |
| 73 | t.Fatal(err) |
| 74 | } |
| 75 | |
| 76 | // Import it into the same graph, with a prefix |
| 77 | if err := g.Import(buf.Bytes(), "imported"); err != nil { |
| 78 | t.Error(err) |
| 79 | } |
| 80 | if err := hasOperations(g, "input", "neg", "imported/input", "imported/neg"); err != nil { |
| 81 | t.Error(err) |
| 82 | } |
| 83 | } |
| 84 | |
| 85 | func TestGraphAddGradients(t *testing.T) { |
| 86 | g := NewGraph() |
nothing calls this directly
no test coverage detected