Searched refs:op_defs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen_test.cc | 110 OpList op_defs; in TEST() local
111 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); // NOLINT in TEST()
112 ApiDefMap api_def_map(op_defs); in TEST()
116 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
123 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
138 OpList op_defs; in TEST() local
139 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); // NOLINT in TEST()
141 ApiDefMap api_def_map(op_defs); in TEST()
145 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
151 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 105 void OpRegistry::GetRegisteredOps(std::vector<OpDef>* op_defs) { in GetRegisteredOps() argument
109 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 | 896 std::vector<tensorflow::OpDef> op_defs; in TF_GetAllOpList() local
897 tensorflow::OpRegistry::Global()->GetRegisteredOps(&op_defs); in TF_GetAllOpList()
899 for (const auto& op : op_defs) { in TF_GetAllOpList()
|