Searched refs:op_defs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen_test.cc | 111 OpList op_defs; in TEST() local
112 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); // NOLINT in TEST()
113 ApiDefMap api_def_map(op_defs); in TEST()
117 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
124 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
139 OpList op_defs; in TEST() local
140 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); // NOLINT in TEST()
142 ApiDefMap api_def_map(op_defs); in TEST()
146 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
152 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen_test.cc | 87 OpList op_defs; in GenerateTsOpFileText() local
89 op_def_str.empty() ? kBaseOpDef : op_def_str, &op_defs); in GenerateTsOpFileText()
90 ApiDefMap api_def_map(op_defs); in GenerateTsOpFileText()
99 WriteTSOps(op_defs, api_def_map, ts_file_path); in GenerateTsOpFileText()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 125 void OpRegistry::GetRegisteredOps(std::vector<OpDef>* op_defs) { in GetRegisteredOps() argument
129 op_defs->push_back(p.second->op_def); in GetRegisteredOps()
|
| D | op.h | 90 void GetRegisteredOps(std::vector<OpDef>* op_defs);
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 998 std::vector<tensorflow::OpDef> op_defs; in TF_GetAllOpList() local
999 tensorflow::OpRegistry::Global()->GetRegisteredOps(&op_defs); in TF_GetAllOpList()
1001 for (const auto& op : op_defs) { in TF_GetAllOpList()
|