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

12

/external/tensorflow/tensorflow/python/framework/
H A Dop_def_registry.py28 def register_op_list(op_list):
30 if not isinstance(op_list, op_def_pb2.OpList):
32 (op_list, type(op_list)))
33 for op_def in op_list.op:
H A Dpython_op_gen_main.cc38 std::vector<string>* op_list) {
54 op_list->emplace_back(op_name.ToString());
72 Status ParseOpListCommandLine(const char* arg, std::vector<string>* op_list) { argument
78 return tensorflow::ReadOpListFromFile(filename, op_list);
80 *op_list = std::move(op_names);
106 void PrintAllPythonOps(const std::vector<string>& op_list, argument
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
37 ReadOpListFromFile(const string& filename, std::vector<string>* op_list) argument
[all...]
H A 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
H A Dop_def_library.py286 def add_op_list(self, op_list):
288 if not isinstance(op_list, op_def_pb2.OpList):
290 (op_list, type(op_list)))
291 for op_def in op_list.op:
H A 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/
H A Dvalidate.h47 const OpList& op_list);
51 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
H A Dvalidate.cc52 const OpList& op_list) {
53 OpListOpRegistry registry(&op_list);
57 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { argument
58 op_registry.Export(false, op_list);
59 RemoveDescriptionsFromOpList(op_list);
51 ValidateGraphDefAgainstOpList(const GraphDef& graph_def, const OpList& op_list) argument
H A Dvalidate_test.cc100 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/core/framework/
H A Dop_gen_lib_test.cc180 OpList op_list;
181 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT
183 ApiDefMap api_map(op_list);
213 OpList op_list;
214 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT
216 ApiDefMap api_map(op_list);
250 OpList op_list;
251 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT
253 ApiDefMap api_map(op_list);
282 OpList op_list;
[all...]
H A Dop_registration_test.cc39 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");
H A Dgraph_def_util_test.cc40 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...]
H A Dload_library.cc31 OpList op_list; member in struct:tensorflow::__anon26245::Library
73 *library.op_list.add_op() = opdef;
94 library.op_list.SerializeToString(&str);
H A Dop.cc84 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) { argument
223 for (const OpDef& op_def : op_list->op()) {
H A Dop_gen_lib.h56 explicit ApiDefMap(const OpList& op_list);
H A Dop_def_util.h70 // Remove all docs from *op_def / *op_list.
72 void RemoveDescriptionsFromOpList(OpList* op_list);
H A Dop.h163 // Does not take ownership of op_list, *op_list must outlive *this.
164 OpListOpRegistry(const OpList* op_list);
/external/tensorflow/tensorflow/python/kernel_tests/
H A 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/c/
H A Dc_api_internal.h71 TF_Buffer op_list; member in struct:TF_Library
165 explicit TF_ApiDefMap(const tensorflow::OpList& op_list) argument
168 api_def_map(op_list),
/external/tensorflow/tensorflow/python/grappler/
H A Ditem_test.py58 op_list = grappler_item.IdentifyImportantOps()
59 self.assertItemsEqual([b'Const', b'Const_1', b'add'], op_list)
/external/tensorflow/tensorflow/python/training/
H A 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)
H A Dsaver.py537 def OpListToDict(op_list, convert_variable_to_tensor=True):
541 op_list: A list, tuple, or set of Variables or SaveableObjects.
551 TypeError: If the type of op_list or its elements is not supported.
554 if not isinstance(op_list, (list, tuple, set)):
556 "list: %s" % op_list)
558 # graph. Sorting the op_list ensures that the resulting graph is always
560 op_list = sorted(op_list, key=lambda x: x.name)
563 for var in op_list:
/external/tensorflow/tensorflow/contrib/model_pruning/python/
H A 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/google-breakpad/src/third_party/libdisasm/swig/
H A Dlibdisasm.i254 x86_op_list *op_list = x86_op_list_new();
257 x86_op_list_append( op_list, &list->op );
260 return op_list;
/external/tensorflow/tensorflow/contrib/learn/python/learn/
H A 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/
H A Dquantize_nodes.cc58 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) {

Completed in 594 milliseconds

12