Home
last modified time | relevance | path

Searched refs:op_list (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dop_gen_lib_test.cc180 OpList op_list; in TEST() local
181 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT in TEST()
183 ApiDefMap api_map(op_list); in TEST()
213 OpList op_list; in TEST() local
214 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT in TEST()
216 ApiDefMap api_map(op_list); in TEST()
250 OpList op_list; in TEST() local
251 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT in TEST()
253 ApiDefMap api_map(op_list); in TEST()
282 OpList op_list; in TEST() local
[all …]
Dgraph_def_util_test.cc40 OpList op_list; in TEST() local
43 op_list.add_op())); in TEST()
44 OpListOpRegistry registry(&op_list); in TEST()
61 OpList op_list; in TEST() local
63 op_list.add_op())); in TEST()
64 OpListOpRegistry registry(&op_list); in TEST()
229 OpList op_list; in TEST() local
231 OpDef* op_def = op_list.add_op(); in TEST()
263 TF_ASSERT_OK(StrippedOpListForGraph(graph_def, OpListOpRegistry(&op_list), in TEST()
286 OpList op_list; in TEST() local
[all …]
Dop.cc84 OpList op_list; in LookUp() local
85 Export(true, &op_list); in LookUp()
88 for (const auto& op : op_list.op()) { in LookUp()
158 OpList op_list; in DebugString() local
159 Export(include_internal, &op_list); in DebugString()
161 for (const auto& op : op_list.op()) { in DebugString()
222 OpListOpRegistry::OpListOpRegistry(const OpList* op_list) { in OpListOpRegistry() argument
223 for (const OpDef& op_def : op_list->op()) { in OpListOpRegistry()
Dop_registration_test.cc39 OpList op_list; in TEST() local
40 registry->Export(true, &op_list); in TEST()
41 EXPECT_EQ(op_list.op().size(), 1); in TEST()
42 EXPECT_EQ(op_list.op(0).name(), "Foo"); in TEST()
Dload_library.cc31 OpList op_list; member
73 *library.op_list.add_op() = opdef; in LoadLibrary()
94 library.op_list.SerializeToString(&str); in LoadLibrary()
Dop_gen_lib.h56 explicit ApiDefMap(const OpList& op_list);
Dop_def_util.h72 void RemoveDescriptionsFromOpList(OpList* op_list);
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_main.cc38 std::vector<string>* op_list) { in ReadOpListFromFile() argument
54 op_list->emplace_back(op_name.ToString()); in ReadOpListFromFile()
72 Status ParseOpListCommandLine(const char* arg, std::vector<string>* op_list) { in ParseOpListCommandLine() argument
78 return tensorflow::ReadOpListFromFile(filename, op_list); in ParseOpListCommandLine()
80 *op_list = std::move(op_names); in ParseOpListCommandLine()
106 void PrintAllPythonOps(const std::vector<string>& op_list, in PrintAllPythonOps() argument
127 std::unordered_set<string> whitelist(op_list.begin(), op_list.end()); in PrintAllPythonOps()
137 PrintEagerPythonOps(ops, api_def_map, op_list, require_shapes, in PrintAllPythonOps()
171 std::vector<tensorflow::string> op_list; in main() local
172 TF_CHECK_OK(tensorflow::ParseOpListCommandLine(argv[2], &op_list)); in main()
[all …]
Dop_def_registry.py28 def register_op_list(op_list): argument
30 if not isinstance(op_list, op_def_pb2.OpList):
32 (op_list, type(op_list)))
33 for op_def in op_list.op:
Dload_library.py61 op_list = op_def_pb2.OpList()
62 op_list.ParseFromString(compat.as_bytes(op_list_str))
79 module.OP_LIST = op_list
Dop_def_library.py286 def add_op_list(self, op_list): argument
288 if not isinstance(op_list, op_def_pb2.OpList):
290 (op_list, type(op_list)))
291 for op_def in op_list.op:
Dmeta_graph_test.py133 op_list = meta_graph.stripped_op_list_for_graph(ops.get_default_graph()
135 self.assertEqual(len(op_list.op), 0)
139 op_list = meta_graph.stripped_op_list_for_graph(ops.get_default_graph()
141 self.assertEqual(["Const", "Square"], [op.name for op in op_list.op])
159 op_list = meta_graph.stripped_op_list_for_graph(graph)
160 self.assertEqual(["Const"], [op.name for op in op_list.op])
/external/tensorflow/tensorflow/core/graph/
Dvalidate_test.cc100 OpList op_list; in TEST() local
101 *op_list.add_op() = op_reg_data.op_def; in TEST()
106 TF_ASSERT_OK(graph::ValidateGraphDefAgainstOpList(graph_def, op_list)); in TEST()
112 OpList op_list; in TEST() local
113 *op_list.add_op() = op_reg_data.op_def; in TEST()
118 ASSERT_FALSE(graph::ValidateGraphDefAgainstOpList(graph_def, op_list).ok()); in TEST()
127 OpList op_list; in TEST() local
128 graph::GetOpListForValidation(&op_list); in TEST()
129 for (const OpDef& op_def : op_list.op()) { in TEST()
Dvalidate.cc52 const OpList& op_list) { in ValidateGraphDefAgainstOpList() argument
53 OpListOpRegistry registry(&op_list); in ValidateGraphDefAgainstOpList()
57 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { in GetOpListForValidation() argument
58 op_registry.Export(false, op_list); in GetOpListForValidation()
59 RemoveDescriptionsFromOpList(op_list); in GetOpListForValidation()
Dvalidate.h47 const OpList& op_list);
51 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
/external/tensorflow/tensorflow/python/kernel_tests/
Drecord_input_test.py120 op_list = sess.run(yield_op)
121 self.assertTrue(len(op_list) is batches)
122 for r in op_list:
174 op_list = sess.run(yield_op)
175 self.assertTrue(len(op_list) is batches)
176 for r in op_list:
/external/tensorflow/tensorflow/python/training/
Dsupervisor.py379 op_list = ops.get_collection(key)
380 if len(op_list) > 1:
382 len(op_list), key)
383 if op_list:
384 return op_list[0]
447 op_list = [
451 if op_list:
452 local_init_op = control_flow_ops.group(*op_list)
Dsaver.py537 def OpListToDict(op_list, convert_variable_to_tensor=True): argument
554 if not isinstance(op_list, (list, tuple, set)):
556 "list: %s" % op_list)
560 op_list = sorted(op_list, key=lambda x: x.name)
563 for var in op_list:
/external/tensorflow/tensorflow/c/
Dc_api_internal.h71 TF_Buffer op_list; member
165 explicit TF_ApiDefMap(const tensorflow::OpList& op_list) in TF_ApiDefMap()
168 api_def_map(op_list), in TF_ApiDefMap()
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning.py236 op_list = []
238 op_list.append(
239 _variable_assign(partition, new_partitioned_values[len(op_list)]))
241 *op_list, name=partitioned_var.name + '_group_assign')
/external/tensorflow/tensorflow/python/grappler/
Ditem_test.py58 op_list = grappler_item.IdentifyImportantOps()
59 self.assertItemsEqual([b'Const', b'Const_1', b'add'], op_list)
/external/google-breakpad/src/third_party/libdisasm/swig/
Dlibdisasm.i254 x86_op_list *op_list = x86_op_list_new(); in x86_insn_op_list() local
257 x86_op_list_append( op_list, &list->op ); in x86_insn_op_list()
260 return op_list; in x86_insn_op_list()
Dlibdisasm_oop.i664 X86_OpList *op_list = new_X86_OpList(); in operand_list() local
667 X86_OpList_append( op_list, &list->op ); in operand_list()
670 return op_list; in operand_list()
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dgraph_actions.py436 op_list = [
440 if op_list:
441 local_init_op = control_flow_ops.group(*op_list)
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc58 static const std::vector<QuantizedOpInfo> op_list = { in GetQuantizedOpList() local
144 return op_list; in GetQuantizedOpList()
644 const std::vector<QuantizedOpInfo>& op_list = GetQuantizedOpList(); in QuantizeNodes() local
648 for (const QuantizedOpInfo& op_info : op_list) { in QuantizeNodes()

12