| /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 | 50 std::unordered_map<string, ApiDef>* name_to_api_def) { in GetGoldenApiDefs()
70 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefsHaveCorrespondingOp()
83 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefInputArgsAreValid()
107 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefOutputArgsAreValid()
131 const OpList& ops, const std::unordered_map<string, ApiDef>& api_defs_map) { in TestAllApiDefAttributeNamesAreValid()
153 const std::unordered_map<string, ApiDef>& api_defs_map) { in TestDeprecatedAttributesSetCorrectly()
180 const std::unordered_map<string, ApiDef>& api_defs_map) { in TestDeprecationVersionSetCorrectly()
206 std::unordered_map<string, ApiDef> api_defs_map_;
302 std::unordered_map<string, ApiDef> api_defs_map_;
|
| D | BUILD | 2 # Provides ApiDef access and ApiDef validation for TensorFlow.
|
| 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 | 272 ApiDef* api_def) { in RenameInDocs()
304 void InitApiDefFromOpDef(const OpDef& op_def, ApiDef* api_def) { in InitApiDefFromOpDef()
306 api_def->set_visibility(ApiDef::VISIBLE); in InitApiDefFromOpDef()
339 void MergeArg(ApiDef::Arg* base_arg, const ApiDef::Arg& new_arg) { in MergeArg()
349 void MergeAttr(ApiDef::Attr* base_attr, const ApiDef::Attr& new_attr) { in MergeAttr()
362 Status MergeApiDefs(ApiDef* base_api_def, const ApiDef& new_api_def) { in MergeApiDefs()
364 if (new_api_def.visibility() != ApiDef::DEFAULT_VISIBILITY) { in MergeApiDefs()
472 ApiDef api_def; in ApiDefMap()
543 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 | 177 const ApiDef::Arg* FindInputArg(StringPiece name, const ApiDef& api_def) { in FindInputArg()
|
| /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_generator.cc | 514 bool CanGenerateOp(const OpDef& op_def, const ApiDef& api_def) { in CanGenerateOp()
515 if (api_def.visibility() == ApiDef::SKIP) { in CanGenerateOp()
546 const ApiDef* api_def = api_map.GetApiDef(op_def.name()); in Run()
|
| D | op_specs.h | 139 static OpSpec Create(const OpDef& op_def, const ApiDef& api_def);
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.h | 71 GenPythonOp(const OpDef& op_def, const ApiDef& api_def,
97 const ApiDef& api_def_;
|
| D | python_op_gen.cc | 107 GenEagerPythonOp(const OpDef& op_def, const ApiDef& api_def, in GenEagerPythonOp()
179 string GetEagerPythonOp(const OpDef& op_def, const ApiDef& api_def, in GetEagerPythonOp()
233 if (api_def_.visibility() == ApiDef::SKIP) { in Code()
363 if (api_def_.visibility() == ApiDef::VISIBLE) { in HandleGraphMode()
639 if (api_def_.visibility() == ApiDef::VISIBLE) { in AddEagerFastPathAndGraphCode()
918 if (api_def_.visibility() != ApiDef::VISIBLE) return; in AddDispatch()
1012 if (api_def->visibility() == ApiDef::SKIP) { in GetPythonOps()
1017 bool is_hidden = api_def->visibility() == ApiDef::HIDDEN; in GetPythonOps()
|
| D | python_op_gen_internal.cc | 487 const ApiDef::Attr* FindAttr(StringPiece name, const ApiDef& api_def) { in FindAttr()
496 GenPythonOp::GenPythonOp(const OpDef& op_def, const ApiDef& api_def, in GenPythonOp()
588 if (api_def_.visibility() != ApiDef::VISIBLE) { in AddExport()
|
| D | c_api_util.py | 124 api_def_proto = api_def_pb2.ApiDef()
|
| /external/tensorflow/tensorflow/go/genop/internal/ |
| D | api_def_map.go | 100 func (m *apiDefMap) Get(opname string) (*pb.ApiDef, error) {
116 apidef = new(pb.ApiDef)
|
| D | genop.go | 132 func generateFunctionForOp(w io.Writer, op *pb.OpDef, apidef *pb.ApiDef) error { argument
383 APIDef *pb.ApiDef
397 func newTmplArgs(op *pb.OpDef, apidef *pb.ApiDef) (*tmplArgs, error) { argument
|
| D | genop_test.go | 30 func GetAPIDef(t *testing.T, opdef *pb.OpDef, apidefText string) *pb.ApiDef {
542 var apidef *pb.ApiDef
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 499 const ApiDef& api_def, in HasOptionalAttrs()
517 explicit OpInfo(const OpDef& graph_op_def, const ApiDef& api_def,
537 const ApiDef& api_def;
543 OpInfo::OpInfo(const OpDef& graph_op_def, const ApiDef& api_def, in OpInfo()
1004 void WriteCCOp(const OpDef& graph_op_def, const ApiDef& api_def, in WriteCCOp()
1140 if (api_def->visibility() == ApiDef::SKIP) continue; in WriteCCOps()
1146 if (api_def->visibility() == ApiDef::HIDDEN) { in WriteCCOps()
|
| /external/tensorflow/tensorflow/java/src/gen/ |
| D | gen_ops.bzl | 36 # Add directory of the first ApiDef source to args.
|
| /external/tensorflow/tensorflow/ |
| D | tensorflow.bzl | 576 # Add directory of the first ApiDef source to args.
769 # Prepare ApiDef directories to pass to the genrule.
775 # Add directory of the first ApiDef source to args.
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2399 tensorflow::ApiDef api_def; in TEST()
2438 tensorflow::ApiDef api_def; in TEST()
|