| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 34 ArgDefs(const OpDef::ArgDef& op_def_arg, const ApiDef::Arg& api_def_arg) in ArgDefs()
38 const ApiDef::Arg& api_def_arg;
43 OpAttrs(const OpDef::AttrDef& op_def_attr, const ApiDef::Attr& api_def_attr) in OpAttrs()
47 const ApiDef::Attr& api_def_attr;
53 GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def);
70 const ApiDef& api_def_;
89 GenTypeScriptOp::GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def) in GenTypeScriptOp()
229 void WriteTSOp(const OpDef& op_def, const ApiDef& api_def, WritableFile* ts) { in WriteTSOp()
282 if (api_def->visibility() == ApiDef::VISIBLE) { in WriteTSOps()
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | api_test.cc | 59 std::unordered_map<string, ApiDef>* name_to_api_def) { in GetGoldenApiDefs()
79 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefsHaveCorrespondingOp()
92 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefInputArgsAreValid()
116 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefOutputArgsAreValid()
140 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefAttributeNamesAreValid()
162 const std::unordered_map<string, ApiDef>& api_defs_map) { in TestDeprecatedAttributesSetCorrectly()
189 const std::unordered_map<string, ApiDef>& api_defs_map) { in TestDeprecationVersionSetCorrectly()
214 std::unordered_map<string, ApiDef> api_defs_map_;
310 std::unordered_map<string, ApiDef> api_defs_map_;
|
| D | BUILD | 2 # Provides ApiDef access and ApiDef validation for TensorFlow.
|
| D | README.md | 1 This folder contains the ApiDef proto definitions of TensorFlow operations.
|
| D | update_api_def.cc | 41 void FillBaseApiDef(ApiDef* api_def, const OpDef& op) { in FillBaseApiDef()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_gen_lib.cc | 274 ApiDef* api_def) { in RenameInDocs()
306 void InitApiDefFromOpDef(const OpDef& op_def, ApiDef* api_def) { in InitApiDefFromOpDef()
308 api_def->set_visibility(ApiDef::VISIBLE); in InitApiDefFromOpDef()
341 void MergeArg(ApiDef::Arg* base_arg, const ApiDef::Arg& new_arg) { in MergeArg()
351 void MergeAttr(ApiDef::Attr* base_attr, const ApiDef::Attr& new_attr) { in MergeAttr()
364 Status MergeApiDefs(ApiDef* base_api_def, const ApiDef& new_api_def) { in MergeApiDefs()
366 if (new_api_def.visibility() != ApiDef::DEFAULT_VISIBILITY) { in MergeApiDefs()
474 ApiDef api_def; in ApiDefMap()
545 const tensorflow::ApiDef* ApiDefMap::GetApiDef(const string& name) const { in GetApiDef()
|
| D | op_gen_lib.h | 92 const ApiDef* GetApiDef(const string& name) const;
95 std::unordered_map<string, ApiDef> map_;
|
| D | op_def_util.h | 53 const ApiDef::Arg* FindInputArg(StringPiece name, const ApiDef& api_def);
|
| D | api_def.proto | 30 message ApiDef { message
44 // different value from another ApiDef.
69 // endpoints are deprecated, set deprecation_message in ApiDef instead.
135 repeated ApiDef op = 1;
|
| D | op_gen_lib_test.cc | 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()
|
| D | op_def_util.cc | 178 const ApiDef::Arg* FindInputArg(StringPiece name, const ApiDef& api_def) { in FindInputArg()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | BUILD | 2 # Provides Java ApiDef access and ApiDef validation for TensorFlow.
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | BUILD | 2 # Provides ApiDef access and ApiDef validation for TensorFlow python.
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| D | op_specs.cc | 286 const ApiDef::Arg& input_api_def, in CreateInput()
302 const ApiDef::Attr& attr_api_def, in CreateAttribute()
321 const ApiDef::Arg& output_api, in CreateOutput()
335 EndpointSpec CreateEndpoint(const OpDef& op_def, const ApiDef& api_def, in CreateEndpoint()
354 OpSpec OpSpec::Create(const OpDef& op_def, const ApiDef& api_def) { in Create()
355 OpSpec op(api_def.graph_op_name(), api_def.visibility() == ApiDef::HIDDEN, in Create()
|
| D | op_specs.h | 139 static OpSpec Create(const OpDef& op_def, const ApiDef& api_def);
|
| D | op_generator.cc | 517 bool CanGenerateOp(const OpDef& op_def, const ApiDef& api_def) { in CanGenerateOp()
518 if (api_def.visibility() == ApiDef::SKIP) { in CanGenerateOp()
549 const ApiDef* api_def = api_map.GetApiDef(op_def.name()); in Run()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.h | 73 GenPythonOp(const OpDef& op_def, const ApiDef& api_def,
99 const ApiDef& api_def_;
|
| D | python_op_gen.cc | 135 GenEagerPythonOp(const OpDef& op_def, const ApiDef& api_def, in GenEagerPythonOp()
218 string GetEagerPythonOp(const OpDef& op_def, const ApiDef& api_def, in GetEagerPythonOp()
274 if (api_def_.visibility() == ApiDef::SKIP) { in Code()
530 if (api_def_.visibility() == ApiDef::VISIBLE) { in HandleGraphMode()
853 if (api_def_.visibility() == ApiDef::VISIBLE) { in AddEagerFastPathAndGraphCode()
1133 if (api_def_.visibility() != ApiDef::VISIBLE) return; in AddDispatch()
1202 if (api_def->visibility() == ApiDef::SKIP) {
1207 bool is_hidden = api_def->visibility() == ApiDef::HIDDEN;
|
| D | python_op_gen_internal.cc | 506 const ApiDef::Attr* FindAttr(StringPiece name, const ApiDef& api_def) { in FindAttr()
515 GenPythonOp::GenPythonOp(const OpDef& op_def, const ApiDef& api_def, in GenPythonOp()
609 if (api_def_.visibility() != ApiDef::VISIBLE) { in AddExport()
|
| D | c_api_util.py | 154 api_def_proto = api_def_pb2.ApiDef()
|
| /external/tensorflow/tensorflow/go/genop/internal/ |
| D | api_def_map.go | 101 func (m *apiDefMap) Get(opname string) (*adpb.ApiDef, error) {
117 apidef = new(adpb.ApiDef)
|
| D | genop.go | 136 func generateFunctionForOp(w io.Writer, op *odpb.OpDef, apidef *adpb.ApiDef) error { argument
387 APIDef *adpb.ApiDef
401 func newTmplArgs(op *odpb.OpDef, apidef *adpb.ApiDef) (*tmplArgs, error) { argument
|
| D | genop_test.go | 31 func GetAPIDef(t *testing.T, opdef *odpb.OpDef, apidefText string) *adpb.ApiDef {
798 var apidef *adpb.ApiDef
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 521 const ApiDef& api_def, in HasOptionalAttrs()
539 explicit OpInfo(const OpDef& graph_op_def, const ApiDef& api_def,
559 const ApiDef& api_def;
565 OpInfo::OpInfo(const OpDef& graph_op_def, const ApiDef& api_def, in OpInfo()
1026 void WriteCCOp(const OpDef& graph_op_def, const ApiDef& api_def, in WriteCCOp()
1162 if (api_def->visibility() == ApiDef::SKIP) continue; in WriteCCOps()
1168 if (api_def->visibility() == ApiDef::HIDDEN) { in WriteCCOps()
|
| /external/tensorflow/tensorflow/java/src/gen/ |
| D | gen_ops.bzl | 34 # Add directory of the first ApiDef source to args.
|