| 276 | } |
| 277 | |
| 278 | FunctionDef XTimesTwoInt32() { |
| 279 | const Tensor kTwo = test::AsScalar<int64>(2); |
| 280 | return FDH::Define( |
| 281 | // Name |
| 282 | "XTimesTwoInt32", |
| 283 | // Args |
| 284 | {"x: int32"}, |
| 285 | // Return values |
| 286 | {"y: int32"}, {}, |
| 287 | // Nodes |
| 288 | { |
| 289 | {{"two"}, "Const", {}, {{"value", kTwo}, {"dtype", DT_INT64}}}, |
| 290 | {{"scale"}, |
| 291 | "Cast", |
| 292 | {"two"}, |
| 293 | {{"SrcT", DT_INT64}, {"DstT", DT_INT32}}}, |
| 294 | {{"y"}, "Mul", {"x", "scale"}, {{"T", DT_INT32}}}, |
| 295 | }); |
| 296 | } |
| 297 | |
| 298 | FunctionDef XTimesFour() { |
| 299 | return FDH::Create( |