Home
last modified time | relevance | path

Searched refs:api_def_map (Results 1 – 16 of 16) 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.cc1124 void WriteCCOps(const OpList& ops, const ApiDefMap& api_def_map, in WriteCCOps() argument
1159 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_test.cc49 ApiDefMap api_def_map(ops); in TEST() local
56 string code = GetPythonOps(ops, api_def_map, {}, "", type_annotate_ops); in TEST()
123 ApiDefMap api_def_map(op_defs); in TEST() local
125 string code = GetPythonOps(op_defs, api_def_map, {}, "", type_annotate_ops); in TEST()
185 ApiDefMap api_def_map(op_defs); in TEST() local
187 string code = GetPythonOps(op_defs, api_def_map, {}, "", type_annotate_ops); in TEST()
244 ApiDefMap api_def_map(op_defs); in TEST() local
246 string code = GetPythonOps(op_defs, api_def_map, {}, "", type_annotate_ops); in TEST()
305 ApiDefMap api_def_map(op_defs); in TEST() local
307 string code = GetPythonOps(op_defs, api_def_map, {}, "", type_annotate_ops); in TEST()
[all …]
Dpython_op_gen_main.cc116 ApiDefMap api_def_map(ops); in PrintAllPythonOps() local
124 TF_CHECK_OK(api_def_map.LoadFileList(env, api_files)); in PrintAllPythonOps()
126 api_def_map.UpdateDocs(); in PrintAllPythonOps()
137 PrintPythonOps(pruned_ops, api_def_map, {}, source_file_name, in PrintAllPythonOps()
140 PrintPythonOps(ops, api_def_map, op_list, source_file_name, in PrintAllPythonOps()
Dpython_op_gen.cc1274 ApiDefMap api_def_map(ops); in GetPythonWrappers() local
1275 return GetPythonOpsImpl(ops, api_def_map, {}); 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.h167 api_def_map(op_list), in TF_ApiDefMap()
173 tensorflow::ApiDefMap api_def_map TF_GUARDED_BY(lock);
Dc_api.cc2423 void TF_ApiDefMapPut(TF_ApiDefMap* api_def_map, const char* text, in TF_ApiDefMapPut() argument
2429 mutex_lock l(api_def_map->lock); in TF_ApiDefMapPut()
2430 if (api_def_map->update_docs_called) { in TF_ApiDefMapPut()
2437 status->status = api_def_map->api_def_map.LoadApiDef(api_def_text); in TF_ApiDefMapPut()
2441 TF_Buffer* TF_ApiDefMapGet(TF_ApiDefMap* api_def_map, const char* name, in TF_ApiDefMapGet() argument
2448 mutex_lock l(api_def_map->lock); in TF_ApiDefMapGet()
2449 if (!api_def_map->update_docs_called) { in TF_ApiDefMapGet()
2450 api_def_map->api_def_map.UpdateDocs(); in TF_ApiDefMapGet()
2451 api_def_map->update_docs_called = true; in TF_ApiDefMapGet()
2454 const auto* api_def = api_def_map->api_def_map.GetApiDef(name_str); in TF_ApiDefMapGet()
Dc_api.h1504 TF_CAPI_EXPORT extern void TF_ApiDefMapPut(TF_ApiDefMap* api_def_map,
1510 TF_CAPI_EXPORT extern TF_Buffer* TF_ApiDefMapGet(TF_ApiDefMap* api_def_map,
Dc_api_test.cc2435 auto* api_def_map = TF_NewApiDefMap(op_list_buf, status); in TEST() local
2442 TF_ApiDefMapGet(api_def_map, op_name.c_str(), op_name.size(), status); in TEST()
2452 TF_DeleteApiDefMap(api_def_map); in TEST()
2463 auto* api_def_map = TF_NewApiDefMap(op_list_buf, status); in TEST() local
2473 TF_ApiDefMapPut(api_def_map, api_def_overwrites.c_str(), in TEST()
2481 TF_ApiDefMapGet(api_def_map, op_name.c_str(), op_name.size(), status); in TEST()
2491 TF_DeleteApiDefMap(api_def_map); in TEST()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc975 [](TF_ApiDefMap* api_def_map, const char* name, size_t name_len) { in PYBIND11_MODULE() argument
981 TF_ApiDefMapGet(api_def_map, name, name_len, status.get()); in PYBIND11_MODULE()
988 [](TF_ApiDefMap* api_def_map, const char* name, size_t name_len) { in PYBIND11_MODULE() argument
993 TF_ApiDefMapPut(api_def_map, name, name_len, status.get()); in PYBIND11_MODULE()