| 332 | } |
| 333 | |
| 334 | FunctionDef WXPlusB() { |
| 335 | return FDH::Define( |
| 336 | // Name |
| 337 | "WXPlusB", |
| 338 | // Args |
| 339 | {"w: T", "x: T", "b: T"}, |
| 340 | // Return values |
| 341 | {"y: T"}, |
| 342 | // Attr def |
| 343 | {"T: {float, double}"}, |
| 344 | // Nodes |
| 345 | {{{"mm"}, |
| 346 | "MatMul", |
| 347 | {"w", "x"}, |
| 348 | {{"T", "$T"}, |
| 349 | {"transpose_a", false}, |
| 350 | {"transpose_b", false}}}, |
| 351 | {{"y"}, "Add", {"mm", "b"}, {{"T", "$T"}}}}); |
| 352 | } |
| 353 | |
| 354 | FunctionDef Swap() { |
| 355 | return FDH::Define( |