Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/framework/
Dop_gen_lib_test.cc181 OpList op_list; in TEST() local
182 protobuf::TextFormat::ParseFromString(kTestOpList, &op_list); // NOLINT in TEST()
184 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
41 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("Project>SomeOp"), op_list.add_op())); in TEST()
42 OpListOpRegistry registry(&op_list); in TEST()
51 OpList op_list; in TEST() local
54 op_list.add_op())); in TEST()
55 OpListOpRegistry registry(&op_list); in TEST()
72 OpList op_list; in TEST() local
74 op_list.add_op())); in TEST()
75 OpListOpRegistry registry(&op_list); in TEST()
239 OpList op_list; in TEST() local
[all …]
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()
Dop.cc126 OpList op_list; in LookUpSlow() local
127 Export(true, &op_list); in LookUpSlow()
130 for (const auto& op : op_list.op()) { in LookUpSlow()
201 OpList op_list; in DebugString() local
202 Export(include_internal, &op_list); in DebugString()
204 for (const auto& op : op_list.op()) { in DebugString()
265 OpListOpRegistry::OpListOpRegistry(const OpList* op_list) { in OpListOpRegistry() argument
266 for (const OpDef& op_def : op_list->op()) { in OpListOpRegistry()
Dload_library.cc31 OpList op_list; member
73 *library.op_list.add_op() = opdef; in LoadDynamicLibrary()
94 library.op_list.SerializeToString(&str); in LoadDynamicLibrary()
Dop_gen_lib.h56 explicit ApiDefMap(const OpList& op_list);
Dop_def_util.h78 void RemoveDescriptionsFromOpList(OpList* op_list);
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_main.cc39 std::vector<string>* op_list) { in ReadOpListFromFile() argument
55 op_list->emplace_back(op_name); in ReadOpListFromFile()
73 Status ParseOpListCommandLine(const char* arg, std::vector<string>* op_list) { in ParseOpListCommandLine() argument
79 return tensorflow::ReadOpListFromFile(filename, op_list); in ParseOpListCommandLine()
81 *op_list = std::move(op_names); in ParseOpListCommandLine()
108 void PrintAllPythonOps(const std::vector<string>& op_list, in PrintAllPythonOps() argument
130 std::unordered_set<string> allowlist(op_list.begin(), op_list.end()); in PrintAllPythonOps()
140 PrintPythonOps(ops, api_def_map, op_list, source_file_name, in PrintAllPythonOps()
177 std::vector<tensorflow::string> op_list; in main() local
178 TF_CHECK_OK(tensorflow::ParseOpListCommandLine(argv[2], &op_list)); in main()
[all …]
Dmeta_graph_test.py135 op_list = meta_graph.stripped_op_list_for_graph(ops.get_default_graph()
137 self.assertEqual(len(op_list.op), 0)
141 op_list = meta_graph.stripped_op_list_for_graph(ops.get_default_graph()
143 self.assertEqual(["Const", "Square"], [op.name for op in op_list.op])
161 op_list = meta_graph.stripped_op_list_for_graph(graph)
162 self.assertEqual(["Const"], [op.name for op in op_list.op])
182 op_list = meta_graph.stripped_op_list_for_graph(graph)
185 [op.name for op in op_list.op])
/external/tensorflow/tensorflow/core/graph/
Dvalidate_test.cc101 OpList op_list; in TEST() local
102 *op_list.add_op() = op_reg_data.op_def; in TEST()
107 TF_ASSERT_OK(graph::ValidateGraphDefAgainstOpList(graph_def, op_list)); in TEST()
113 OpList op_list; in TEST() local
114 *op_list.add_op() = op_reg_data.op_def; in TEST()
119 ASSERT_FALSE(graph::ValidateGraphDefAgainstOpList(graph_def, op_list).ok()); in TEST()
128 OpList op_list; in TEST() local
129 graph::GetOpListForValidation(&op_list); in TEST()
130 for (const OpDef& op_def : op_list.op()) { in TEST()
Dvalidate.cc54 const OpList& op_list) { in ValidateGraphDefAgainstOpList() argument
55 OpListOpRegistry registry(&op_list); in ValidateGraphDefAgainstOpList()
59 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { in GetOpListForValidation() argument
60 op_registry.Export(false, op_list); in GetOpListForValidation()
61 RemoveDescriptionsFromOpList(op_list); in GetOpListForValidation()
Dvalidate.h48 const OpList& op_list);
52 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_fusion.cc167 explicit ShapeConstraintAnalysis(const SmallVectorImpl<Operation*>& op_list) { in ShapeConstraintAnalysis() argument
168 PropagateEquality(op_list); in ShapeConstraintAnalysis()
179 void PropagateEquality(const SmallVectorImpl<Operation*>& op_list) { in PropagateEquality() argument
189 for (Operation* op : op_list) { in PropagateEquality()
239 explicit FusionPlanner(const SmallVectorImpl<Operation*>& op_list) in FusionPlanner() argument
240 : op_list_(op_list), in FusionPlanner()
241 shape_analysis_(op_list), in FusionPlanner()
242 cycle_detector_(op_list.size()) { in FusionPlanner()
277 const SmallVectorImpl<Operation*>& op_list() const { return op_list_; } in op_list() function in mlir::mhlo::__anonbfa6fffa0111::FusionPlanner
284 const SmallVectorImpl<Operation*>& op_list = planner->op_list(); in Cluster() local
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Drecord_input_test.py122 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/python/training/saving/
Dsaveable_object_util.py226 def op_list_to_dict(op_list, convert_variable_to_tensor=True): argument
243 if not isinstance(op_list, (list, tuple, set)):
245 "list: %s" % op_list)
247 op_list = nest.flatten(list(op_list))
251 op_list = sorted(op_list, key=lambda x: x.name)
254 for var in op_list:
/external/tensorflow/tensorflow/c/
Dc_api_internal.h67 TF_Buffer op_list; member
164 explicit TF_ApiDefMap(const tensorflow::OpList& op_list) in TF_ApiDefMap()
167 api_def_map(op_list), in TF_ApiDefMap()
Dops_test.cc43 ::tensorflow::OpList op_list; in TEST()
44 op_list.ParseFromArray(op_list_buffer->data, op_list_buffer->length); in TEST()
46 for (const auto& op : op_list.op()) { in TEST()
150 ::tensorflow::OpList op_list; in TEST()
151 op_list.ParseFromArray(op_list_buffer->data, op_list_buffer->length); in TEST()
153 for (const auto& op : op_list.op()) { in TEST()
/external/tensorflow/tensorflow/python/training/
Dsupervisor.py381 op_list = ops.get_collection(key)
382 if len(op_list) > 1:
384 len(op_list), key)
385 if op_list:
386 return op_list[0]
448 op_list = [
452 if op_list:
453 local_init_op = control_flow_ops.group(*op_list)
/external/tensorflow/tensorflow/lite/toco/logging/
Dgen_html.py167 for i in range(len(toco_conversion_log_before.op_list)):
169 pre_op_profile += "<tr><td>" + toco_conversion_log_before.op_list[
179 for op in toco_conversion_log_after.op_list:
Dgen_html_test.py38 toco_conversion_log_before.op_list.extend([
44 toco_conversion_log_after.op_list.extend([
/external/tensorflow/tensorflow/core/grappler/utils/
Dpattern_utils.cc41 std::vector<string> op_list = str_util::Split(pattern.op, '|'); in DoesOpTypePatternMatch() local
42 for (const string& op : op_list) { in DoesOpTypePatternMatch()
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_generator.h47 Status Run(const OpList& op_list, const string& base_package,
Dop_generator.cc531 Status OpGenerator::Run(const OpList& op_list, const string& base_package, in Run() argument
533 ApiDefMap api_map(op_list); in Run()
536 for (const auto& op : op_list.op()) { in Run()
548 for (const auto& op_def : op_list.op()) { in Run()
/external/tensorflow/tensorflow/python/grappler/
Ditem_test.py59 op_list = grappler_item.IdentifyImportantOps()
60 self.assertItemsEqual(['Const', 'Const_1', '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()

12