| 452 | } |
| 453 | |
| 454 | FunctionDef LessThanOrEqualToN(int64 N) { |
| 455 | const Tensor kN = test::AsScalar<int64>(N); |
| 456 | return FDH::Define( |
| 457 | // Name |
| 458 | "LessThanOrEqualToN", |
| 459 | // Args |
| 460 | {"x: T"}, |
| 461 | // Return values |
| 462 | {"z: bool"}, |
| 463 | // Attr def |
| 464 | {"T: {float, double, int32, int64}"}, |
| 465 | // Nodes |
| 466 | { |
| 467 | {{"N"}, "Const", {}, {{"value", kN}, {"dtype", DT_INT64}}}, |
| 468 | {{"y"}, "Cast", {"N"}, {{"SrcT", DT_INT64}, {"DstT", "$T"}}}, |
| 469 | {{"z"}, "LessEqual", {"x", "y"}, {{"T", "$T"}}}, |
| 470 | }); |
| 471 | } |
| 472 | |
| 473 | FunctionDef XPlusOneXTimesY() { |
| 474 | const Tensor kOne = test::AsScalar<int64>(1); |