/external/tensorflow/tensorflow/python/framework/ |
H A D | op_def_registry.py | 28 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 D | 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) { 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 D | load_library.py | 61 op_list = op_def_pb2.OpList() 62 op_list.ParseFromString(compat.as_bytes(op_list_str)) 79 module.OP_LIST = op_list
|
H A D | op_def_library.py | 286 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 D | meta_graph_test.py | 133 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 D | validate.h | 47 const OpList& op_list); 51 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
|
H A D | validate.cc | 52 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 D | 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/core/framework/ |
H A D | op_gen_lib_test.cc | 180 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 D | 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");
|
H A D | 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...] |
H A D | load_library.cc | 31 OpList op_list; member in struct:tensorflow::__anon26245::Library 73 *library.op_list.add_op() = opdef; 94 library.op_list.SerializeToString(&str);
|
H A D | 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) { argument 223 for (const OpDef& op_def : op_list->op()) {
|
H A D | op_gen_lib.h | 56 explicit ApiDefMap(const OpList& op_list);
|
H A D | op_def_util.h | 70 // Remove all docs from *op_def / *op_list. 72 void RemoveDescriptionsFromOpList(OpList* op_list);
|
H A D | op.h | 163 // 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 D | record_input_test.py | 120 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 D | c_api_internal.h | 71 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 D | item_test.py | 58 op_list = grappler_item.IdentifyImportantOps() 59 self.assertItemsEqual([b'Const', b'Const_1', b'add'], op_list)
|
/external/tensorflow/tensorflow/python/training/ |
H A D | supervisor.py | 379 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 D | saver.py | 537 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 D | pruning.py | 236 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 D | libdisasm.i | 254 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 D | graph_actions.py | 436 op_list = [ 440 if op_list: 441 local_init_op = control_flow_ops.group(*op_list)
|
/external/tensorflow/tensorflow/tools/graph_transforms/ |
H A D | 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) {
|