Lines Matching defs:Builder
42 NodeDefBuilder& Builder() {
43 EXPECT_FALSE(op_def_.name().empty()) << "Must call Op() before Builder()";
50 void ExpectSuccess(const NodeDefBuilder& builder,
54 Status status = builder.Finalize(&node_def);
64 InOutTypesForNode(node_def, builder.op_def(), &in_types, &out_types);
78 void ExpectFailures(const NodeDefBuilder& builder,
81 Status status = builder.Finalize(&node_def);
92 void ExpectFailure(const NodeDefBuilder& builder, const string& message) {
93 ExpectFailures(builder, {message});
98 void ExpectInvalid(const NodeDefBuilder& builder, const string& message) {
100 Status status = builder.Finalize(&node_def);
118 ExpectSuccess(Builder().Input("x", 0, DT_INT32), {DT_INT32}, {DT_FLOAT},
122 ExpectSuccess(Builder().Input("y", 2, DT_INT32), {DT_INT32}, {DT_FLOAT},
126 ExpectSuccess(Builder().Input(FakeInput()), {DT_INT32}, {DT_FLOAT}, R"proto(
129 ExpectSuccess(Builder().Input(FakeInput(DT_INT32)), {DT_INT32}, {DT_FLOAT},
133 ExpectSuccess(Builder().Input(FakeInput(DT_INT32_REF)), {DT_INT32},
138 Builder().ControlInput("x").Input(FakeInput()).ControlInput("y"),
143 ExpectSuccess(Builder().Input(FakeInput()).Device("ddd"), {DT_INT32},
148 ExpectFailure(Builder().Input("x", 0, DT_INT32).Input("y", 0, DT_INT32),
154 ExpectFailure(Builder(), "0 inputs specified of 1 inputs in Op while");
157 NodeDefBuilder& builder = Builder();
159 TF_EXPECT_OK(builder.Input(FakeInput()).Finalize(nullptr));
161 ExpectSuccess(builder, {DT_INT32}, {DT_FLOAT}, R"proto(
166 NodeDefBuilder& builder = Builder();
168 ExpectFailure(builder, "0 inputs specified of 1 inputs in Op while");
169 builder.Input(FakeInput());
171 ExpectSuccess(builder, {DT_INT32}, {DT_FLOAT}, R"proto(
174 builder.Input(FakeInput(DT_INT32));
175 ExpectFailure(builder, "More Input() calls than the 1 input_args while");
179 ExpectFailure(Builder().Input("x", 0, DT_FLOAT),
182 ExpectFailure(Builder().Input("x", 0, DT_FLOAT_REF),
186 ExpectFailure(Builder().Input(FakeInput(3, DT_FLOAT)),
189 ExpectFailure(Builder().Input(FakeInput(3)),
193 ExpectInvalid(Builder().Input(FakeInput()).ControlInput("z:2"),
197 ExpectFailure(Builder().Input("", 0, DT_INT32),
200 ExpectFailure(Builder().Input("^x", 0, DT_INT32),
205 NodeDefBuilder builder("n", "Op Does Not Exist");
206 builder.Input(FakeInput())
211 ExpectFailures(builder, {"Op type not registered 'Op Does Not Exist'",
221 ExpectSuccess(Builder().Input(FakeInput(DT_INT32)), {DT_INT32}, {DT_INT32},
226 ExpectSuccess(Builder().Input(FakeInput(DT_FLOAT)), {DT_FLOAT}, {DT_FLOAT},
232 ExpectSuccess(Builder().Input(FakeInput(DT_BOOL)).Attr("T", DT_BOOL),
238 ExpectFailure(Builder().Input(FakeInput(DT_BOOL)).Attr("T", DT_STRING),
241 ExpectFailure(Builder().Attr("T", DT_STRING).Input(FakeInput(DT_BOOL)),
244 ExpectFailure(Builder().Attr("T", 12).Input(FakeInput(DT_BOOL)),
251 ExpectSuccess(Builder().Attr("T", DT_INT32), {}, {DT_INT32}, R"proto(
255 ExpectSuccess(Builder().Attr("T", DT_FLOAT), {}, {DT_FLOAT}, R"proto(
260 ExpectSuccess(Builder().Attr("T", DT_FLOAT).Attr("T", DT_FLOAT), {},
266 ExpectFailure(Builder().Attr("T", DT_BOOL).Attr("T", DT_FLOAT),
270 ExpectInvalid(Builder(), "NodeDef missing attr 'T' from");
274 Builder().Attr("T", {DT_INT32, DT_BOOL}),
277 ExpectInvalid(Builder().Attr("T", 12),
286 ExpectSuccess(Builder(), {}, {DT_STRING}, R"proto(
290 ExpectSuccess(Builder().Attr("T", DT_BOOL), {}, {DT_BOOL}, R"proto(
299 ExpectSuccess(Builder().Input(FakeInput(DT_INT32)).Input(FakeInput(DT_INT32)),
304 ExpectSuccess(Builder().Input(FakeInput(DT_STRING)).Input(FakeInput()),
310 ExpectFailure(Builder().Input(FakeInput(DT_BOOL)).Input(FakeInput(DT_STRING)),
319 ExpectSuccess(Builder().Input(FakeInput(DT_STRING)), {DT_STRING}, {DT_STRING},
324 ExpectInvalid(Builder().Input(FakeInput(DT_INT32)),
332 ExpectSuccess(Builder().Input(FakeInput({DT_STRING, DT_INT32})),
338 ExpectSuccess(Builder().Input(FakeInput(3, DT_BOOL)),
344 ExpectInvalid(Builder().Input(FakeInput(0)),
347 ExpectInvalid(Builder().Input(FakeInput({})),
350 ExpectInvalid(Builder().Input(FakeInput(DT_BOOL)),
353 ExpectFailures(Builder().Input(FakeInput()),
363 ExpectSuccess(Builder().Input(FakeInput(0)), {}, {}, R"proto(
366 ExpectSuccess(Builder().Input(FakeInput(DataTypeVector())), {}, {}, R"proto(
369 ExpectSuccess(Builder().Input(FakeInput({})), {}, {}, R"proto(
372 ExpectSuccess(Builder().Input(FakeInput({DT_BOOL})), {DT_BOOL}, {}, R"proto(
383 ExpectSuccess(Builder()
391 Builder().Input(FakeInput({DT_INT32, DT_BOOL})).Input(FakeInput()),
396 ExpectSuccess(Builder().Input(FakeInput(0)).Input(FakeInput(0)), {}, {},
400 ExpectSuccess(Builder().Input(FakeInput(0)).Input(FakeInput()), {}, {},
404 ExpectFailure(Builder()
414 ExpectSuccess(Builder().Attr("T", {DT_FLOAT}), {}, {DT_FLOAT}, R"proto(
418 ExpectSuccess(Builder().Attr("T", {DT_STRING, DT_BOOL}), {},
423 ExpectSuccess(Builder().Attr("T", DataTypeVector()), {}, {}, R"proto(
428 Builder().Attr("T", DT_FLOAT),
437 ExpectSuccess(Builder().Input(FakeInput({DT_STRING, DT_BOOL})),
442 ExpectInvalid(Builder().Input(FakeInput({DT_STRING, DT_INT32})),
452 ExpectSuccess(Builder().Attr("t", {DT_BOOL, DT_STRING}), {},
457 ExpectInvalid(Builder().Attr("t", {DT_STRING, DT_INT32}),
465 ExpectSuccess(Builder().Attr("a", 12), {}, {}, R"proto(
469 ExpectInvalid(Builder().Attr("a", "bad"),
473 Builder().Attr("a", {12}),
477 ExpectInvalid(Builder(), "NodeDef missing attr 'a' from Op<");
480 ExpectInvalid(Builder().Attr("b", 12),
484 ExpectInvalid(Builder().Attr("a", 12).Attr("extra", 12),
491 ExpectSuccess(Builder().Attr("a", 1.2f /* float */), {}, {}, R"proto(
495 ExpectSuccess(Builder().Attr("a", 1.2 /* double */), {}, {}, R"proto(
500 ExpectInvalid(Builder().Attr("a", 12),
507 ExpectSuccess(Builder().Attr("a", {true, false, true}), {}, {}, R"proto(
512 ExpectSuccess(Builder().Attr("a", std::vector<bool>()), {}, {}, R"proto(
517 ExpectInvalid(Builder().Attr("a", {0}),
525 ExpectSuccess(Builder().Attr("a", 12), {}, {}, R"proto(
528 ExpectInvalid(Builder().Attr("a", 2),
535 ExpectSuccess(Builder().Attr("a", {1, 2}), {}, {}, R"proto(
539 ExpectInvalid(Builder().Attr("a", {17}),
546 ExpectSuccess(Builder().Attr("a", "oranges"), {}, {}, R"proto(
551 Builder().Attr("a", "invalid"),
559 ExpectSuccess(Builder().Attr("a", {"oranges", "apples"}), {}, {}, R"proto(
564 Builder().Attr("a", {"apples", "invalid", "oranges"}),
572 ExpectSuccess(Builder().Attr("a", TensorShape({5})), {}, {}, R"proto(
576 ExpectSuccess(Builder().Attr("a", TensorShape({4, 3, 2})), {}, {}, R"proto(
581 ExpectSuccess(Builder().Attr("a", TensorShape({3, 2})), {}, {},
587 ExpectSuccess(Builder().Attr("a", TensorShape()), {}, {}, R"proto(
595 ExpectSuccess(Builder(), {}, {}, R"proto(
599 ExpectSuccess(Builder().Attr("a", "kiwi"), {}, {}, R"proto(
609 ExpectSuccess(Builder(), {}, {}, R"proto(
619 ExpectSuccess(Builder().Attr("c", "strawberry"), {}, {}, R"proto(
635 ExpectSuccess(Builder()
656 ExpectSuccess(Builder(), {}, {}, R"proto(
660 ExpectSuccess(Builder().Attr("a", {3}), {}, {}, R"proto(
664 ExpectSuccess(Builder().Attr("a", std::vector<int>()), {}, {}, R"proto(
672 ExpectSuccess(Builder(), {}, {}, R"proto(
676 ExpectSuccess(Builder().Attr("a", {3}), {}, {}, R"proto(
680 ExpectSuccess(Builder().Attr("a", std::vector<int>()), {}, {}, R"proto(
688 ExpectSuccess(Builder().Input(FakeInput(2)), {DT_INT32, DT_INT32}, {},
693 ExpectSuccess(Builder().Input(FakeInput(5, DT_INT32)),
699 ExpectFailures(Builder().Input(FakeInput(2, DT_STRING)),
703 ExpectInvalid(Builder().Input(FakeInput(1)),
707 Builder().Input(FakeInput(DT_INT32)),
712 ExpectFailure(Builder().Input({{"in", 0, DT_INT32}, {"in", 1, DT_STRING}}),
716 Builder().Input(FakeInput()),
728 ExpectSuccess(Builder().Input(FakeInput(2, DT_INT32)), {DT_INT32, DT_INT32},
734 ExpectSuccess(Builder().Input(FakeInput(3, DT_STRING)),
742 Builder().Input(FakeInput(2)),
747 ExpectFailure(Builder().Input(FakeInput({DT_INT32, DT_STRING})),
750 ExpectFailure(Builder().Input({{"in", 0, DT_INT32}, {"in", 1, DT_STRING}}),
753 ExpectInvalid(Builder().Input(FakeInput(1, DT_INT32)),
756 ExpectFailure(Builder().Input("in", 0, DT_INT32),
766 ExpectSuccess(Builder().Input(FakeInput(2, DT_BOOL)), {DT_BOOL, DT_BOOL}, {},
772 ExpectSuccess(Builder().Input(FakeInput(3, DT_STRING)),
779 ExpectInvalid(Builder().Input(FakeInput(2, DT_INT32)),
790 ExpectSuccess(Builder().Input(FakeInput(2)).Input(FakeInput(2)),
796 ExpectSuccess(Builder().Input(FakeInput(0)).Input(FakeInput()), {}, {},
800 ExpectFailure(Builder().Input(FakeInput(3)).Input(FakeInput(1)),
811 ExpectSuccess(Builder().Input(FakeInput(2, DT_INT32)).Input(FakeInput()),
819 Builder().Input(FakeInput(3, DT_INT32)).Input(FakeInput(1, DT_INT32)),
822 ExpectFailure(Builder().Input(FakeInput(3, DT_INT32)).Input(FakeInput(1)),
826 Builder().Input(FakeInput(2, DT_INT32)).Input(FakeInput(2, DT_STRING)),
830 Builder().Input(FakeInput(2, DT_INT32)).Input(FakeInput(DT_STRING)),
843 Builder().Input(FakeInput(2, DT_INT32)).Input(FakeInput(2, DT_BOOL)),
852 Builder().Input(FakeInput(2, DT_INT32)).Input(FakeInput(DT_BOOL)),
861 Builder().Input(FakeInput(3, DT_INT32)).Input(FakeInput(1, DT_STRING)),
874 Builder().Input(FakeInput(1, DT_INT32)).Input(FakeInput(3, DT_INT32)),
882 ExpectSuccess(Builder
889 ExpectSuccess(Builder().Input(FakeInput(0)).Input(FakeInput(1, DT_BOOL)),
897 Builder().Input(FakeInput(2, DT_INT32)).Input(FakeInput(2, DT_STRING)),
904 ExpectSuccess(Builder().Attr("N", 2), {}, {DT_INT32, DT_INT32}, R"proto(
908 ExpectSuccess(Builder().Attr("N", 3), {}, {DT_INT32, DT_INT32, DT_INT32},
913 ExpectInvalid(Builder().Attr("N", 1),
917 Builder().Attr("N", {3}),
920 ExpectInvalid(Builder(), "NodeDef missing attr 'N' from");
928 ExpectSuccess(Builder(), {}, {DT_INT32, DT_INT32, DT_INT32}, R"proto(
932 ExpectSuccess(Builder().Attr("N", 2), {}, {DT_INT32, DT_INT32}, R"proto(
943 ExpectSuccess(Builder().Attr("T", DT_INT32).Attr("N", 2), {},
949 ExpectSuccess(Builder().Attr("N", 3).Attr("T", DT_STRING), {},
955 ExpectInvalid(Builder().Attr("N", 1).Attr("T", DT_STRING),
959 Builder().Attr("N", 3).Attr("T", {DT_STRING}),
969 ExpectSuccess(Builder(), {}, {DT_BOOL, DT_BOOL}, R"proto(
974 ExpectSuccess(Builder().Attr("N", 3), {}, {DT_BOOL, DT_BOOL, DT_BOOL},
980 ExpectSuccess(Builder().Attr("T", DT_INT32), {}, {DT_INT32, DT_INT32},
986 ExpectSuccess(Builder().Attr("N", 3).Attr("T", DT_INT32), {},
999 ExpectSuccess(Builder().Attr("N", 3).Attr("T", DT_BOOL), {},
1005 ExpectInvalid(Builder().Attr("N", 3).Attr("T", DT_INT32),
1013 ExpectSuccess(Builder().Input(FakeInput(DT_INT32_REF)), {DT_INT32_REF}, {},
1017 ExpectFailure(Builder().Input(FakeInput(DT_BOOL_REF)),
1020 ExpectFailure(Builder().Input(FakeInput(DT_INT32)),
1027 ExpectSuccess(Builder().Input(FakeInput(DT_BOOL_REF)), {DT_BOOL_REF}, {},
1032 ExpectFailure(Builder().Input(FakeInput(DT_BOOL)),
1039 ExpectSuccess(Builder(), {}, {DT_STRING_REF}, R"proto(
1046 ExpectSuccess(Builder().Attr("t", DT_BOOL), {}, {DT_BOOL_REF}, R"proto(
1054 ExpectSuccess(Builder().Device("ADevice"), {}, {}, R"proto(
