| /external/tensorflow/tensorflow/core/framework/ |
| op_registration_test.cc | 39 OpList op_list; local
40 registry->Export(true, &op_list);
41 EXPECT_EQ(op_list.op().size(), 1);
42 EXPECT_EQ(op_list.op(0).name(), "Foo");
|
| load_library.cc | 31 OpList op_list; member in struct:tensorflow::__anon39566::Library
73 *library.op_list.add_op() = opdef;
94 library.op_list.SerializeToString(&str);
|
| graph_def_util_test.cc | 40 OpList op_list; local
43 op_list.add_op()));
44 OpListOpRegistry registry(&op_list);
61 OpList op_list; local
63 op_list.add_op()));
64 OpListOpRegistry registry(&op_list);
229 OpList op_list; local
231 OpDef* op_def = op_list.add_op();
263 TF_ASSERT_OK(StrippedOpListForGraph(graph_def, OpListOpRegistry(&op_list),
286 OpList op_list; local
[all...] |
| op.cc | 84 OpList op_list; local
85 Export(true, &op_list);
88 for (const auto& op : op_list.op()) {
158 OpList op_list; local
159 Export(include_internal, &op_list);
161 for (const auto& op : op_list.op()) {
222 OpListOpRegistry::OpListOpRegistry(const OpList* op_list) {
223 for (const OpDef& op_def : op_list->op()) {
|
| /external/tensorflow/tensorflow/core/graph/ |
| validate_test.cc | 100 OpList op_list; local
101 *op_list.add_op() = op_reg_data.op_def;
106 TF_ASSERT_OK(graph::ValidateGraphDefAgainstOpList(graph_def, op_list));
112 OpList op_list; local
113 *op_list.add_op() = op_reg_data.op_def;
118 ASSERT_FALSE(graph::ValidateGraphDefAgainstOpList(graph_def, op_list).ok());
127 OpList op_list; local
128 graph::GetOpListForValidation(&op_list);
129 for (const OpDef& op_def : op_list.op()) {
|
| /external/tensorflow/tensorflow/python/framework/ |
| python_op_gen_main.cc | 38 std::vector<string>* op_list) {
54 op_list->emplace_back(op_name.ToString());
72 Status ParseOpListCommandLine(const char* arg, std::vector<string>* op_list) {
78 return tensorflow::ReadOpListFromFile(filename, op_list);
80 *op_list = std::move(op_names);
106 void PrintAllPythonOps(const std::vector<string>& op_list,
127 std::unordered_set<string> whitelist(op_list.begin(), op_list.end());
137 PrintEagerPythonOps(ops, api_def_map, op_list, require_shapes,
171 std::vector<tensorflow::string> op_list; local
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_internal.h | 71 TF_Buffer op_list; member in struct:TF_Library
165 explicit TF_ApiDefMap(const tensorflow::OpList& op_list)
168 api_def_map(op_list),
|
| c_api_test.cc | 198 tensorflow::OpList op_list; local
199 EXPECT_TRUE(op_list.ParseFromArray(op_list_buf.data, op_list_buf.length));
200 ASSERT_EQ(op_list.op_size(), 1);
201 EXPECT_EQ("TestCApi", op_list.op(0).name());
332 tensorflow::OpList op_list; local
333 EXPECT_TRUE(op_list.ParseFromArray(buf->data, buf->length));
334 EXPECT_GT(op_list.op_size(), 0);
[all...] |
| c_api.cc | 879 library_filename, &lib_handle->lib_handle, &lib_handle->op_list.data,
880 &lib_handle->op_list.length);
888 TF_Buffer TF_GetOpList(TF_Library* lib_handle) { return lib_handle->op_list; }
891 tensorflow::port::Free(const_cast<void*>(lib_handle->op_list.data));
898 tensorflow::OpList op_list; local
900 *(op_list.add_op()) = op;
903 TF_CHECK_OK(MessageToBuffer(op_list, ret));
2686 tensorflow::OpList op_list; local
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes.cc | 58 static const std::vector<QuantizedOpInfo> op_list = { local
144 return op_list;
644 const std::vector<QuantizedOpInfo>& op_list = GetQuantizedOpList(); local
648 for (const QuantizedOpInfo& op_info : op_list) {
[all...] |
