Home
last modified time | relevance | path

Searched refs:api_def_map (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dcc_op_gen_test.cc87 const ApiDefMap& api_def_map, string* h_file_text, in GenerateCcOpFiles() argument
96 WriteCCOps(ops, api_def_map, h_file_path, cc_file_path); in GenerateCcOpFiles()
113 ApiDefMap api_def_map(op_defs); in TEST() local
117 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
123 TF_ASSERT_OK(api_def_map.LoadApiDef(api_def)); in TEST()
124 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
142 ApiDefMap api_def_map(op_defs); in TEST() local
146 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
151 TF_ASSERT_OK(api_def_map.LoadApiDef(api_def)); in TEST()
152 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
[all …]
Dcc_op_gen_main.cc35 ApiDefMap api_def_map(ops); in PrintAllCCOps() local
44 TF_CHECK_OK(api_def_map.LoadFile(env, api_def_file_pattern)); in PrintAllCCOps()
50 api_def_map.UpdateDocs(); in PrintAllCCOps()
52 WriteCCOps(ops, api_def_map, dot_h, dot_cc); in PrintAllCCOps()
Dcc_op_gen.h26 void WriteCCOps(const OpList& ops, const ApiDefMap& api_def_map,
Dcc_op_gen.cc1102 void WriteCCOps(const OpList& ops, const ApiDefMap& api_def_map, in WriteCCOps() argument
1137 const auto* api_def = api_def_map.GetApiDef(graph_op_def.name()); in WriteCCOps()
/external/tensorflow/tensorflow/python/framework/
Dc_api_util_test.py29 api_def_map = c_api_util.ApiDefMap()
30 self.assertIn("Add", api_def_map.op_names())
33 api_def_map = c_api_util.ApiDefMap()
34 op_def = api_def_map.get_op_def("Add")
36 api_def = api_def_map.get_api_def("Add")
40 api_def_map = c_api_util.ApiDefMap()
51 api_def_map.put_api_def(api_def_text)
52 api_def = api_def_map.get_api_def("Add")
Dpython_op_gen_main.cc115 ApiDefMap api_def_map(ops); in PrintAllPythonOps() local
123 TF_CHECK_OK(api_def_map.LoadFileList(env, api_files)); in PrintAllPythonOps()
125 api_def_map.UpdateDocs(); in PrintAllPythonOps()
136 PrintPythonOps(pruned_ops, api_def_map, {}, require_shapes, in PrintAllPythonOps()
139 PrintPythonOps(ops, api_def_map, op_list, require_shapes, source_file_name); in PrintAllPythonOps()
Dpython_op_gen.cc1097 ApiDefMap api_def_map(ops); in GetPythonWrappers() local
1098 return GetPythonOps(ops, api_def_map, {}, false); in GetPythonWrappers()
/external/tensorflow/tensorflow/js/ops/
Dts_op_gen_test.cc90 ApiDefMap api_def_map(op_defs); in GenerateTsOpFileText() local
93 TF_ASSERT_OK(api_def_map.LoadApiDef(api_def_str)); in GenerateTsOpFileText()
99 WriteTSOps(op_defs, api_def_map, ts_file_path); in GenerateTsOpFileText()
Dts_op_gen.h26 void WriteTSOps(const OpList& ops, const ApiDefMap& api_def_map,
Dts_op_gen.cc265 void WriteTSOps(const OpList& ops, const ApiDefMap& api_def_map, in WriteTSOps() argument
281 const auto* api_def = api_def_map.GetApiDef(op_def.name()); in WriteTSOps()
/external/tensorflow/tensorflow/c/
Dc_api_internal.h174 api_def_map(op_list), in TF_ApiDefMap()
180 tensorflow::ApiDefMap api_def_map GUARDED_BY(lock);
Dc_api.cc2831 void TF_ApiDefMapPut(TF_ApiDefMap* api_def_map, const char* text, in TF_ApiDefMapPut() argument
2837 mutex_lock l(api_def_map->lock); in TF_ApiDefMapPut()
2838 if (api_def_map->update_docs_called) { in TF_ApiDefMapPut()
2845 status->status = api_def_map->api_def_map.LoadApiDef(api_def_text); in TF_ApiDefMapPut()
2849 TF_Buffer* TF_ApiDefMapGet(TF_ApiDefMap* api_def_map, const char* name, in TF_ApiDefMapGet() argument
2856 mutex_lock l(api_def_map->lock); in TF_ApiDefMapGet()
2857 if (!api_def_map->update_docs_called) { in TF_ApiDefMapGet()
2858 api_def_map->api_def_map.UpdateDocs(); in TF_ApiDefMapGet()
2859 api_def_map->update_docs_called = true; in TF_ApiDefMapGet()
2862 const auto* api_def = api_def_map->api_def_map.GetApiDef(name_str); in TF_ApiDefMapGet()
Dc_api_test.cc2388 auto* api_def_map = TF_NewApiDefMap(op_list_buf, status); in TEST() local
2395 TF_ApiDefMapGet(api_def_map, op_name.c_str(), op_name.size(), status); in TEST()
2405 TF_DeleteApiDefMap(api_def_map); in TEST()
2416 auto* api_def_map = TF_NewApiDefMap(op_list_buf, status); in TEST() local
2426 TF_ApiDefMapPut(api_def_map, api_def_overwrites.c_str(), in TEST()
2434 TF_ApiDefMapGet(api_def_map, op_name.c_str(), op_name.size(), status); in TEST()
2444 TF_DeleteApiDefMap(api_def_map); in TEST()
Dc_api.h1708 TF_CAPI_EXPORT extern void TF_ApiDefMapPut(TF_ApiDefMap* api_def_map,
1714 TF_CAPI_EXPORT extern TF_Buffer* TF_ApiDefMapGet(TF_ApiDefMap* api_def_map,