gorgonia.org/gorgonia@v0.9.17/bench_typesystem_test.go (about)

     1  package gorgonia
     2  
     3  import "testing"
     4  
     5  func BenchmarkTypeSystem(b *testing.B) {
     6  	g := NewGraph()
     7  	x := NewTensor(g, Float64, 2, WithName("x"), WithShape(10, 10))
     8  	y := NewTensor(g, Float64, 2, WithName("y"), WithShape(10, 10))
     9  	op := newEBOByType(addOpType, Float64, Float64)
    10  	for i := 0; i < b.N; i++ {
    11  		inferNodeType(op, x, y)
    12  	}
    13  }