/external/tensorflow/tensorflow/cc/framework/ |
D | cc_op_gen_test.cc | 87 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 …]
|
D | cc_op_gen_main.cc | 35 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()
|
D | cc_op_gen.h | 26 void WriteCCOps(const OpList& ops, const ApiDefMap& api_def_map,
|
D | cc_op_gen.cc | 1102 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/ |
D | c_api_util_test.py | 29 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")
|
D | python_op_gen_main.cc | 115 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()
|
D | python_op_gen.cc | 1097 ApiDefMap api_def_map(ops); in GetPythonWrappers() local 1098 return GetPythonOps(ops, api_def_map, {}, false); in GetPythonWrappers()
|
/external/tensorflow/tensorflow/js/ops/ |
D | ts_op_gen_test.cc | 90 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()
|
D | ts_op_gen.h | 26 void WriteTSOps(const OpList& ops, const ApiDefMap& api_def_map,
|
D | ts_op_gen.cc | 265 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/ |
D | c_api_internal.h | 174 api_def_map(op_list), in TF_ApiDefMap() 180 tensorflow::ApiDefMap api_def_map GUARDED_BY(lock);
|
D | c_api.cc | 2831 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()
|
D | c_api_test.cc | 2388 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()
|
D | c_api.h | 1708 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,
|