| /external/tensorflow/tensorflow/core/framework/ |
| D | op_gen_lib_test.cc | 217 const auto* api_def = api_map.GetApiDef("testop"); in TEST() local
218 ASSERT_EQ(expected_api_def, api_def->DebugString()); in TEST()
255 const auto* api_def = api_map.GetApiDef("testop"); in TEST() local
256 EXPECT_EQ(1, api_def->endpoint_size()); in TEST()
257 EXPECT_EQ("testop1", api_def->endpoint(0).name()); in TEST()
260 *api_defs.add_op() = *api_def; in TEST()
287 auto* api_def = api_map.GetApiDef("testop"); in TEST() local
288 EXPECT_EQ(ApiDef::VISIBLE, api_def->visibility()); in TEST()
293 EXPECT_EQ(ApiDef::VISIBLE, api_def->visibility()); in TEST()
298 EXPECT_EQ(ApiDef::HIDDEN, api_def->visibility()); in TEST()
[all …]
|
| D | op_gen_lib.cc | 272 ApiDef* api_def) { in RenameInDocs() argument
275 for (int i = 0; i < api_def->in_arg_size(); ++i) { in RenameInDocs()
276 if (!api_def->in_arg(i).description().empty()) { in RenameInDocs()
278 api_def->mutable_in_arg(i)->mutable_description()); in RenameInDocs()
281 for (int i = 0; i < api_def->out_arg_size(); ++i) { in RenameInDocs()
282 if (!api_def->out_arg(i).description().empty()) { in RenameInDocs()
284 api_def->mutable_out_arg(i)->mutable_description()); in RenameInDocs()
287 for (int i = 0; i < api_def->attr_size(); ++i) { in RenameInDocs()
288 if (!api_def->attr(i).description().empty()) { in RenameInDocs()
290 api_def->mutable_attr(i)->mutable_description()); in RenameInDocs()
[all …]
|
| D | op_def_util.h | 53 const ApiDef::Arg* FindInputArg(StringPiece name, const ApiDef& api_def);
|
| D | op_def_util.cc | 177 const ApiDef::Arg* FindInputArg(StringPiece name, const ApiDef& api_def) { in FindInputArg() argument
178 for (int i = 0; i < api_def.in_arg_size(); ++i) { in FindInputArg()
179 if (api_def.in_arg(i).name() == name) { in FindInputArg()
180 return &api_def.in_arg(i); in FindInputArg()
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | api_test.cc | 89 const auto& api_def = api_def_iter->second; in TestAllApiDefInputArgsAreValid() local
90 for (const auto& api_def_arg : api_def.in_arg()) { in TestAllApiDefInputArgsAreValid()
113 const auto& api_def = api_def_iter->second; in TestAllApiDefOutputArgsAreValid() local
114 for (const auto& api_def_arg : api_def.out_arg()) { in TestAllApiDefOutputArgsAreValid()
137 const auto& api_def = api_def_iter->second; in TestAllApiDefAttributeNamesAreValid() local
138 for (const auto& api_def_attr : api_def.attr()) { in TestAllApiDefAttributeNamesAreValid()
156 const auto& api_def = name_and_api_def.second; in TestDeprecatedAttributesSetCorrectly() local
157 for (const auto& endpoint : api_def.endpoint()) { in TestDeprecatedAttributesSetCorrectly()
164 ASSERT_TRUE(api_def.deprecation_message().empty() || in TestDeprecatedAttributesSetCorrectly()
171 ASSERT_NE(num_deprecated_endpoints, api_def.endpoint_size()) in TestDeprecatedAttributesSetCorrectly()
[all …]
|
| D | update_api_def.cc | 41 void FillBaseApiDef(ApiDef* api_def, const OpDef& op) { in FillBaseApiDef() argument
42 api_def->set_graph_op_name(op.name()); in FillBaseApiDef()
46 auto* api_def_in_arg = api_def->add_in_arg(); in FillBaseApiDef()
53 auto* api_def_out_arg = api_def->add_out_arg(); in FillBaseApiDef()
61 auto* api_def_attr = api_def->add_attr(); in FillBaseApiDef()
67 api_def->set_summary(op.summary()); in FillBaseApiDef()
68 api_def->set_description(op.description()); in FillBaseApiDef()
|
| D | update_api_def.sh | 25 bazel build //tensorflow/core/api_def:update_api_def
26 bazel-bin/tensorflow/core/api_def/update_api_def \
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 499 const ApiDef& api_def, in HasOptionalAttrs() argument
501 for (int i = 0; i < api_def.attr_size(); ++i) { in HasOptionalAttrs()
502 const auto& attr(api_def.attr(i)); in HasOptionalAttrs()
517 explicit OpInfo(const OpDef& graph_op_def, const ApiDef& api_def,
537 const ApiDef& api_def; member
543 OpInfo::OpInfo(const OpDef& graph_op_def, const ApiDef& api_def, in OpInfo() argument
545 : graph_op_def(graph_op_def), api_def(api_def), aliases(aliases) { in OpInfo()
546 op_name = api_def.endpoint(0).name(); in OpInfo()
548 has_optional_attrs = HasOptionalAttrs(api_def, inferred_input_attrs); in OpInfo()
553 if (!api_def.summary().empty()) { in OpInfo()
[all …]
|
| D | cc_op_gen_test.cc | 104 const string api_def = R"( in TEST() local
123 TF_ASSERT_OK(api_def_map.LoadApiDef(api_def)); in TEST()
131 const string api_def = R"( in TEST() local
151 TF_ASSERT_OK(api_def_map.LoadApiDef(api_def)); in TEST()
158 const string api_def = R"( in TEST() local
184 TF_ASSERT_OK(api_def_map.LoadApiDef(api_def)); in TEST()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | c_api_util_test.py | 36 api_def = api_def_map.get_api_def("Add")
37 self.assertEqual(api_def.graph_op_name, "Add")
52 api_def = api_def_map.get_api_def("Add")
53 self.assertEqual(api_def.graph_op_name, "Add")
54 self.assertEqual(api_def.summary, "Returns x + y element-wise.")
|
| D | python_op_gen.cc | 107 GenEagerPythonOp(const OpDef& op_def, const ApiDef& api_def, in GenEagerPythonOp() argument
109 : python_op_gen_internal::GenPythonOp(op_def, api_def, function_name) { in GenEagerPythonOp()
179 string GetEagerPythonOp(const OpDef& op_def, const ApiDef& api_def, in GetEagerPythonOp() argument
181 return GenEagerPythonOp(op_def, api_def, function_name).Code(); in GetEagerPythonOp()
1010 const auto* api_def = api_defs.GetApiDef(op_def.name()); in GetPythonOps() local
1012 if (api_def->visibility() == ApiDef::SKIP) { in GetPythonOps()
1017 bool is_hidden = api_def->visibility() == ApiDef::HIDDEN; in GetPythonOps()
1051 GetEagerPythonOp(op_def, *api_def, function_name)); in GetPythonOps()
|
| D | python_op_gen_internal.cc | 487 const ApiDef::Attr* FindAttr(StringPiece name, const ApiDef& api_def) { in FindAttr() argument
488 for (int i = 0; i < api_def.attr_size(); ++i) { in FindAttr()
489 if (api_def.attr(i).name() == name) { in FindAttr()
490 return &api_def.attr(i); in FindAttr()
496 GenPythonOp::GenPythonOp(const OpDef& op_def, const ApiDef& api_def, in GenPythonOp() argument
499 api_def_(api_def), in GenPythonOp()
|
| D | python_op_gen_internal.h | 71 GenPythonOp(const OpDef& op_def, const ApiDef& api_def,
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 53 GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def);
89 GenTypeScriptOp::GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def) in GenTypeScriptOp() argument
90 : op_def_(op_def), api_def_(api_def), num_outputs_(0) {} in GenTypeScriptOp()
229 void WriteTSOp(const OpDef& op_def, const ApiDef& api_def, WritableFile* ts) { in WriteTSOp() argument
230 GenTypeScriptOp ts_op(op_def, api_def); in WriteTSOp()
231 TF_CHECK_OK(ts->Append(GenTypeScriptOp(op_def, api_def).Code())); in WriteTSOp()
281 const auto* api_def = api_def_map.GetApiDef(op_def.name()); in WriteTSOps() local
282 if (api_def->visibility() == ApiDef::VISIBLE) { in WriteTSOps()
283 WriteTSOp(op_def, *api_def, ts_file.get()); in WriteTSOps()
|
| D | ts_op_gen_test.cc | 115 const string api_def = R"pb( in TEST() local
120 GenerateTsOpFileText("", api_def, &ts_file_text); in TEST()
129 const string api_def = R"( in TEST() local
137 GenerateTsOpFileText("", api_def, &ts_file_text); in TEST()
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| D | op_specs.cc | 335 EndpointSpec CreateEndpoint(const OpDef& op_def, const ApiDef& api_def, in CreateEndpoint() argument
348 Javadoc::Create(ParseDocumentation(api_def.summary())) in CreateEndpoint()
349 .details(ParseDocumentation(api_def.description()))); in CreateEndpoint()
354 OpSpec OpSpec::Create(const OpDef& op_def, const ApiDef& api_def) { in Create() argument
355 OpSpec op(api_def.graph_op_name(), api_def.visibility() == ApiDef::HIDDEN, in Create()
358 for (const string& next_input_name : api_def.arg_order()) { in Create()
361 op.inputs_.push_back(CreateInput(op_def.input_arg(i), api_def.in_arg(i), in Create()
372 CreateAttribute(op_def.attr(i), api_def.attr(i), &type_resolver); in Create()
383 CreateOutput(op_def.output_arg(i), api_def.out_arg(i), &type_resolver)); in Create()
385 for (const auto& endpoint_def : api_def.endpoint()) { in Create()
[all …]
|
| D | op_generator.cc | 514 bool CanGenerateOp(const OpDef& op_def, const ApiDef& api_def) { in CanGenerateOp() argument
515 if (api_def.visibility() == ApiDef::SKIP) { in CanGenerateOp()
546 const ApiDef* api_def = api_map.GetApiDef(op_def.name()); in Run() local
547 if (CanGenerateOp(op_def, *api_def)) { in Run()
548 OpSpec op(OpSpec::Create(op_def, *api_def)); in Run()
|
| D | op_specs.h | 139 static OpSpec Create(const OpDef& op_def, const ApiDef& api_def);
|
| /external/tensorflow/tensorflow/python/ |
| D | build_defs.bzl | 34 "//tensorflow/core/api_def:base_api_def",
35 "//tensorflow/core/api_def:python_api_def",
|
| /external/tensorflow/tensorflow/cc/ |
| D | BUILD | 475 api_def_srcs = ["//tensorflow/core/api_def:base_api_def"],
484 api_def_srcs = ["//tensorflow/core/api_def:base_api_def"],
493 api_def_srcs = ["//tensorflow/core/api_def:base_api_def"],
631 "//tensorflow/core/api_def:base_api_def",
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_pb_text_files.txt | 4 tensorflow/core/framework/api_def.pb_text.cc
|
| D | proto_text_pb_cc_files.txt | 4 tensorflow/core/framework/api_def.pb.cc
|
| D | tf_proto_files.txt | 8 tensorflow/core/framework/api_def.proto
|
| D | proto_text_pb_h_files.txt | 4 tensorflow/core/framework/api_def.pb.h
|
| /external/tensorflow/tensorflow/java/ |
| D | BUILD | 76 "//tensorflow/core/api_def:base_api_def",
77 "//tensorflow/core/api_def:java_api_def",
|