| 489 | } |
| 490 | |
| 491 | FunctionDef XYXLessThanOrEqualToN(int64 N) { |
| 492 | const Tensor kN = test::AsScalar<int64>(N); |
| 493 | return FDH::Define( |
| 494 | // Name |
| 495 | "XYXLessThanOrEqualToN", |
| 496 | // Args |
| 497 | {"x: T", "y: T"}, |
| 498 | // Return values |
| 499 | {"z: bool"}, |
| 500 | // Attr def |
| 501 | {"T: {float, double, int32, int64}"}, |
| 502 | // Nodes |
| 503 | { |
| 504 | {{"N"}, "Const", {}, {{"value", kN}, {"dtype", DT_INT64}}}, |
| 505 | {{"N1"}, "Cast", {"N"}, {{"SrcT", DT_INT64}, {"DstT", "$T"}}}, |
| 506 | {{"z"}, "LessEqual", {"x", "N1"}, {{"T", "$T"}}}, |
| 507 | }); |
| 508 | } |
| 509 | |
| 510 | FunctionDef RandomUniformLess() { |
| 511 | const Tensor kZero = test::AsScalar<int32>(0); |