| /external/tensorflow/tensorflow/core/framework/ |
| D | op_def_util_test.cc | 46 Status TestBuilder(const OpDefBuilder& builder) { in TestBuilder()
70 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Attr("a: int"))); in TEST_F()
71 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Input("a: int32"))); in TEST_F()
72 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Output("a: bool"))); in TEST_F()
73 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Attr("t: type").Input("a: t"))); in TEST_F()
74 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Attr("a: int = 3"))); in TEST_F()
75 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Attr("a: int >= -5"))); in TEST_F()
76 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Attr("a: int >= -5"))); in TEST_F()
77 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Attr("a: int >= -5 = 3"))); in TEST_F()
78 TF_EXPECT_OK(TestBuilder(OpDefBuilder("X").Attr("a: numbertype"))); in TEST_F()
[all …]
|
| D | op_def_builder.h | 53 class OpDefBuilder {
56 explicit OpDefBuilder(std::string op_name);
89 OpDefBuilder& Attr(std::string spec);
106 OpDefBuilder& Input(std::string spec);
107 OpDefBuilder& Output(std::string spec);
111 OpDefBuilder& SetIsCommutative();
112 OpDefBuilder& SetIsAggregate();
113 OpDefBuilder& SetIsStateful();
114 OpDefBuilder& SetAllowsUninitializedInput();
117 OpDefBuilder& Deprecated(int version, std::string explanation);
[all …]
|
| D | op_compatibility_test.cc | 224 OpDefBuilder("AddAttr").Output("ndef: string").Finalize(&old_op)); in TEST_F()
236 TF_ASSERT_OK(OpDefBuilder("LessStrict") in TEST_F()
254 TF_ASSERT_OK(OpDefBuilder("RemoveRestriction") in TEST_F()
272 TF_ASSERT_OK(OpDefBuilder("AttrOrder") in TEST_F()
295 TF_ASSERT_OK(OpDefBuilder("TypePolymorphic") in TEST_F()
317 TF_ASSERT_OK(OpDefBuilder("MakeList") in TEST_F()
340 TF_ASSERT_OK(OpDefBuilder("MakePolyList") in TEST_F()
362 TF_ASSERT_OK(OpDefBuilder("MakeAnyList") in TEST_F()
384 TF_ASSERT_OK(OpDefBuilder("PolyIntoList") in TEST_F()
410 TF_ASSERT_OK(OpDefBuilder("MakeMultipleSameList") in TEST_F()
[all …]
|
| D | node_def_builder_test.cc | 35 void Op(const OpDefBuilder& op_def_builder) { in Op()
119 Op(OpDefBuilder("Simple").Input("a: int32").Output("out: float")); in TEST_F()
219 Op(OpDefBuilder("Polymorphic") in TEST_F()
252 Op(OpDefBuilder("PolymorphicOut").Output("out: T").Attr("T: type")); in TEST_F()
285 Op(OpDefBuilder("PolymorphicDefaultOut") in TEST_F()
299 Op(OpDefBuilder("Binary").Input("a: T").Input("b: T").Output("out: T").Attr( in TEST_F()
318 Op(OpDefBuilder("Restrict") in TEST_F()
333 Op(OpDefBuilder("TypeList").Input("a: T").Attr("T: list(type)")); in TEST_F()
364 Op(OpDefBuilder("TypeListNoMin").Input("a: T").Attr("T: list(type) >= 0")); in TEST_F()
381 Op(OpDefBuilder("TypeListTwice") in TEST_F()
[all …]
|
| D | op_def_builder.cc | 560 OpDefBuilder::OpDefBuilder(string op_name) { in OpDefBuilder() function in tensorflow::OpDefBuilder
564 OpDefBuilder& OpDefBuilder::Attr(string spec) { in Attr()
569 OpDefBuilder& OpDefBuilder::Input(string spec) { in Input()
574 OpDefBuilder& OpDefBuilder::Output(string spec) { in Output()
579 OpDefBuilder& OpDefBuilder::ControlOutput(string name) { in ControlOutput()
585 OpDefBuilder& OpDefBuilder::Doc(string text) { in Doc()
596 OpDefBuilder& OpDefBuilder::SetIsCommutative() { in SetIsCommutative()
601 OpDefBuilder& OpDefBuilder::SetIsAggregate() { in SetIsAggregate()
606 OpDefBuilder& OpDefBuilder::SetIsStateful() { in SetIsStateful()
611 OpDefBuilder& OpDefBuilder::SetAllowsUninitializedInput() { in SetAllowsUninitializedInput()
[all …]
|
| D | graph_def_util_test.cc | 31 Status FinalizeOpDef(const OpDefBuilder& b, OpDef* op_def) { in FinalizeOpDef()
41 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("Project>SomeOp"), op_list.add_op())); in TEST()
53 FinalizeOpDef(OpDefBuilder("NoChangeWithDefault").Attr("a: int = 12"), in TEST()
73 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("NoChangeNoDefault").Attr("a: int"), in TEST()
97 FinalizeOpDef(OpDefBuilder("UsesDefault"), consumer_op_list.add_op())); in TEST()
101 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("UsesDefault").Attr("a: int = 17"), in TEST()
128 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("ChangedFromDefault"), in TEST()
134 FinalizeOpDef(OpDefBuilder("ChangedFromDefault").Attr("a: int = 17"), in TEST()
158 FinalizeOpDef(OpDefBuilder("Underscore"), consumer_op_list.add_op())); in TEST()
163 FinalizeOpDef(OpDefBuilder("Underscore"), producer_op_list.add_op())); in TEST()
[all …]
|
| D | node_def_util_test.cc | 34 OpDef ToOpDef(const OpDefBuilder& builder) { in ToOpDef()
78 const OpDef op = ToOpDef(OpDefBuilder("In").Input("i: T").Attr("T: type")); in TEST()
143 ToOpDef(OpDefBuilder("Out").Output("o: T").Attr("T: numbertype")); in TEST()
163 const OpDef op = ToOpDef(OpDefBuilder("Enum").Attr("e: {'apple','orange'}")); in TEST()
186 const OpDef op = ToOpDef(OpDefBuilder("SameIn") in TEST()
218 ToOpDef(OpDefBuilder("AnyIn").Input("i: T").Attr("T: list(type) >= 1")); in TEST()
245 const OpDef op_def1 = ToOpDef(OpDefBuilder("None")); in TEST()
252 const OpDef op_def2 = ToOpDef(OpDefBuilder("WithAttr").Attr("v: int")); in TEST()
410 const OpDef op_def = ToOpDef(OpDefBuilder("Simple") in TEST()
431 const OpDef op_def = ToOpDef(OpDefBuilder("Simple") in TEST()
[all …]
|
| D | op_def_builder_test.cc | 41 OpDefBuilder b() { return OpDefBuilder("Test"); } in b()
43 void ExpectSuccess(const OpDefBuilder& builder, StringPiece proto, in ExpectSuccess()
64 void ExpectOrdered(const OpDefBuilder& builder, StringPiece proto) { in ExpectOrdered()
77 void ExpectFailure(const OpDefBuilder& builder, const string& error) { in ExpectFailure()
|
| D | node_properties_test.cc | 25 OpDef ToOpDef(const OpDefBuilder& builder) { in ToOpDef()
34 : op_reg_(ToOpDef(OpDefBuilder("Foo") in MockOpRegistry()
|
| D | op.h | 267 const ::tensorflow::OpDefBuilder& builder() const { return builder_; } in builder()
272 mutable ::tensorflow::OpDefBuilder builder_;
|
| D | common_shape_fns_test.cc | 39 OpDefBuilder b("dummy"); in MakeOpDef()
54 TF_CHECK_OK(OpDefBuilder("Assert") in TEST()
74 TF_CHECK_OK(OpDefBuilder("L2Loss") in TEST()
102 TF_CHECK_OK(OpDefBuilder("MatMul") in TEST()
409 TF_CHECK_OK(OpDefBuilder("BiasAdd") in TEST()
539 TF_CHECK_OK(OpDefBuilder("BiasAddGrad") in TEST()
|
| D | function.cc | 1820 OpDefBuilder b(function_name); in Create()
1877 OpDefBuilder b(name); in Define()
|
| D | shape_inference_test.cc | 42 OpDefBuilder b("dummy"); in MakeOpDefWithLists()
112 OpDefBuilder b("dummy"); in MakeOpDef()
|
| /external/tensorflow/tensorflow/c/ |
| D | ops.cc | 26 using ::tensorflow::OpDefBuilder;
36 auto* result = new OpDefBuilder(op_name); in TF_NewOpDefinitionBuilder()
41 delete reinterpret_cast<OpDefBuilder*>(builder); in TF_DeleteOpDefinitionBuilder()
46 reinterpret_cast<OpDefBuilder*>(builder)->Input(input_spec); in TF_OpDefinitionBuilderAddInput()
51 reinterpret_cast<OpDefBuilder*>(builder)->Output(output_spec); in TF_OpDefinitionBuilderAddOutput()
57 reinterpret_cast<OpDefBuilder*>(builder)->func_name(); \
67 reinterpret_cast<OpDefBuilder*>(builder)->Attr(attr_spec); in DEFINE_BUILDER_BOOL_SETTER()
72 reinterpret_cast<OpDefBuilder*>(builder)->Deprecated(version, explanation); in TF_OpDefinitionBuilderDeprecated()
77 auto* cc_builder = reinterpret_cast<OpDefBuilder*>(builder); in TF_RegisterOpDefinition()
91 auto* cc_builder = reinterpret_cast<OpDefBuilder*>(builder); in TF_OpDefinitionBuilderSetShapeInferenceFunction()
|
| D | ops_test.cc | 178 OpDefBuilder b("dummy"); in MakeOpDef()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | validate_test.cc | 100 TF_ASSERT_OK(OpDefBuilder("UniqueSnowflake").Finalize(&op_reg_data)); in TEST()
112 TF_ASSERT_OK(OpDefBuilder("NotAnywhere").Finalize(&op_reg_data)); in TEST()
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | update_api_def.cc | 126 OpDefBuilder b(op.name()); in ValidateOpDocs()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel_test.cc | 358 tensorflow::OpDefBuilder b("dummy"); in MakeOpDef()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_info.cc | 211 OpDefBuilder op_def_builder(api_name_); in InitializeFromParamSpecs()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 2473 OpDefBuilder b("dummy"); in TEST()
|