/external/tensorflow/tensorflow/python/framework/ |
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:
|
python_op_gen_main.cc | 39 std::vector<string>* op_list) { 55 op_list->emplace_back(op_name); 73 Status ParseOpListCommandLine(const char* arg, std::vector<string>* op_list) { 79 return tensorflow::ReadOpListFromFile(filename, op_list); 81 *op_list = std::move(op_names); 108 void PrintAllPythonOps(const std::vector<string>& op_list, 129 std::unordered_set<string> whitelist(op_list.begin(), op_list.end()); 139 PrintPythonOps(ops, api_def_map, op_list, require_shapes, source_file_name); 172 std::vector<tensorflow::string> op_list; local [all...] |
load_library.py | 64 op_list = op_def_pb2.OpList() 65 op_list.ParseFromString(compat.as_bytes(op_list_str)) 82 module.OP_LIST = op_list
|
op_def_library.py | 302 def add_op_list(self, op_list): 304 if not isinstance(op_list, op_def_pb2.OpList): 306 (op_list, type(op_list))) 307 for op_def in op_list.op:
|
/external/tensorflow/tensorflow/core/graph/ |
validate.h | 48 const OpList& op_list); 52 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
|
validate_test.cc | 101 OpList op_list; local 102 *op_list.add_op() = op_reg_data.op_def; 107 TF_ASSERT_OK(graph::ValidateGraphDefAgainstOpList(graph_def, op_list)); 113 OpList op_list; local 114 *op_list.add_op() = op_reg_data.op_def; 119 ASSERT_FALSE(graph::ValidateGraphDefAgainstOpList(graph_def, op_list).ok()); 128 OpList op_list; local 129 graph::GetOpListForValidation(&op_list); 130 for (const OpDef& op_def : op_list.op()) {
|
validate.cc | 54 const OpList& op_list) { 55 OpListOpRegistry registry(&op_list); 59 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { 60 op_registry.Export(false, op_list); 61 RemoveDescriptionsFromOpList(op_list);
|
/external/tensorflow/tensorflow/core/framework/ |
op_gen_lib_test.cc | 181 OpList op_list; 182 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT 184 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...] |
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");
|
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...] |
load_library.cc | 31 OpList op_list; member in struct:tensorflow::__anon44971::Library 73 *library.op_list.add_op() = opdef; 94 library.op_list.SerializeToString(&str);
|
op.cc | 100 OpList op_list; local 101 Export(true, &op_list); 104 for (const auto& op : op_list.op()) { 187 OpList op_list; local 188 Export(include_internal, &op_list); 190 for (const auto& op : op_list.op()) { 251 OpListOpRegistry::OpListOpRegistry(const OpList* op_list) { 252 for (const OpDef& op_def : op_list->op()) {
|
op_gen_lib.h | 56 explicit ApiDefMap(const OpList& op_list);
|
op_def_util.h | 75 // Remove all docs from *op_def / *op_list. 77 void RemoveDescriptionsFromOpList(OpList* op_list);
|
op.h | 169 // Does not take ownership of op_list, *op_list must outlive *this. 170 OpListOpRegistry(const OpList* op_list);
|
/external/tensorflow/tensorflow/java/src/gen/cc/ |
op_generator.h | 47 Status Run(const OpList& op_list, const string& base_package,
|
op_generator.cc | 528 Status OpGenerator::Run(const OpList& op_list, const string& base_package, 530 ApiDefMap api_map(op_list); 532 // Only load api files that correspond to the requested "op_list" 533 for (const auto& op : op_list.op()) { 545 for (const auto& op_def : op_list.op()) {
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
record_input_test.py | 122 op_list = self.evaluate(yield_op) 123 self.assertTrue(len(op_list) is batches) 124 for r in op_list: 179 op_list = self.evaluate(yield_op) 180 self.assertTrue(len(op_list) is batches) 181 for r in op_list:
|
/external/tensorflow/tensorflow/c/ |
c_api_internal.h | 75 TF_Buffer op_list; member in struct:TF_Library 171 explicit TF_ApiDefMap(const tensorflow::OpList& op_list) 174 api_def_map(op_list),
|
/external/tensorflow/tensorflow/python/training/saving/ |
saveable_object_util.py | 210 def op_list_to_dict(op_list, convert_variable_to_tensor=True): 214 op_list: A list, tuple, or set of Variables or SaveableObjects. 224 TypeError: If the type of op_list or its elements is not supported. 227 if not isinstance(op_list, (list, tuple, set)): 229 "list: %s" % op_list) 231 # graph. Sorting the op_list ensures that the resulting graph is always 233 op_list = sorted(op_list, key=lambda x: x.name) 236 for var in op_list:
|
/external/tensorflow/tensorflow/contrib/mpi_collectives/ |
mpi_ops.py | 30 def _load_library(name, op_list=None): 35 op_list: A list of names of operators that the library should have. If None 44 for expected_op in (op_list or []): 45 for lib_op in library.OP_LIST.op:
|
/external/tensorflow/tensorflow/contrib/model_pruning/python/ |
pruning_utils.py | 254 op_list = [] 256 op_list.append( 257 variable_assign(partition, new_partitioned_values[len(op_list)])) 259 *op_list, name=partitioned_var.name + '_group_assign')
|
/external/tensorflow/tensorflow/python/grappler/ |
item_test.py | 59 op_list = grappler_item.IdentifyImportantOps() 60 self.assertItemsEqual(['Const', 'Const_1', 'add'], op_list)
|
/external/tensorflow/tensorflow/python/training/ |
supervisor.py | 383 op_list = ops.get_collection(key) 384 if len(op_list) > 1: 386 len(op_list), key) 387 if op_list: 388 return op_list[0] 451 op_list = [ 455 if op_list: 456 local_init_op = control_flow_ops.group(*op_list) [all...] |
/external/google-breakpad/src/third_party/libdisasm/swig/ |
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;
|