MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TestOperationShapeAttribute

Function TestOperationShapeAttribute

tensorflow/go/operation_test.go:86–99  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

84}
85
86func TestOperationShapeAttribute(t *testing.T) {
87 g := NewGraph()
88 _, err := g.AddOperation(OpSpec{
89 Type: "Placeholder",
90 Attrs: map[string]interface{}{
91 "dtype": Float,
92 "shape": MakeShape(-1, 3),
93 },
94 })
95 if err != nil {
96 t.Fatal(err)
97 }
98 // If and when the API to get attributes is added, check that here.
99}
100
101func TestOutputDataTypeAndShape(t *testing.T) {
102 graph := NewGraph()

Callers

nothing calls this directly

Calls 3

AddOperationMethod · 0.95
NewGraphFunction · 0.85
MakeShapeFunction · 0.85

Tested by

no test coverage detected