| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_test.cc | 120 OpList op_defs; in TEST() local
121 OpRegistry::Global()->Export(false, &op_defs); in TEST()
122 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); in TEST()
123 ApiDefMap api_def_map(op_defs); in TEST()
125 string code = GetPythonOps(op_defs, api_def_map, {}, "", type_annotate_ops); in TEST()
182 OpList op_defs; in TEST() local
183 OpRegistry::Global()->Export(false, &op_defs); in TEST()
184 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); in TEST()
185 ApiDefMap api_def_map(op_defs); in TEST()
187 string code = GetPythonOps(op_defs, api_def_map, {}, "", type_annotate_ops); in TEST()
[all …]
|
| D | meta_graph.py | 193 op_defs = []
197 op_defs.append(op_def)
198 return op_def_pb2.OpList(op=op_defs)
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen_test.cc | 111 OpList op_defs; in TEST() local
112 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); // NOLINT in TEST()
113 ApiDefMap api_def_map(op_defs); in TEST()
117 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
124 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
139 OpList op_defs; in TEST() local
140 protobuf::TextFormat::ParseFromString(kBaseOpDef, &op_defs); // NOLINT in TEST()
142 ApiDefMap api_def_map(op_defs); in TEST()
146 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
152 GenerateCcOpFiles(env, op_defs, api_def_map, &h_file_text, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen_test.cc | 87 OpList op_defs; in GenerateTsOpFileText() local
89 op_def_str.empty() ? kBaseOpDef : op_def_str, &op_defs); in GenerateTsOpFileText()
90 ApiDefMap api_def_map(op_defs); in GenerateTsOpFileText()
99 WriteTSOps(op_defs, api_def_map, ts_file_path); in GenerateTsOpFileText()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tools/ |
| D | tflite_op_coverage_spec_getters_gen.cc | 314 std::vector<Record *> op_defs = records.getAllDerivedDefinitions("TFL_Op"); in TFLiteOpCoverageSpecWritersMain() local
315 EmitStaticQuantOp(op_defs, &os); in TFLiteOpCoverageSpecWritersMain()
316 EmitDynamicRangeOp(op_defs, &os); in TFLiteOpCoverageSpecWritersMain()
317 EmitStaticQuantWithInt16ActOp(op_defs, &os); in TFLiteOpCoverageSpecWritersMain()
318 EmitSparseOp(op_defs, &os); in TFLiteOpCoverageSpecWritersMain()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_factory_test.cc | 145 std::vector<OpDef> op_defs; in TEST() local
146 OpRegistry::Global()->GetRegisteredOps(&op_defs); in TEST()
151 for (const OpDef& op_def : op_defs) { in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 362 def __init__(self, op_defs): argument
364 self._op_defs = op_defs
642 def __init__(self, ctx, op_defs): argument
646 self._op_defs = op_defs
1495 def __init__(self, op_defs): argument
1496 self._op_defs = op_defs
1516 def tfr_gen(func, op_defs): argument
1518 mlir_code, _ = TfrGen(op_defs).transform(func, None)
1523 def tfr_funcs_gen_from_module(source, op_defs, method_prefix=None, argument
1562 mlir_funcs = [tfr_gen(func, op_defs) for func in py_funcs]
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 140 void OpRegistry::GetRegisteredOps(std::vector<OpDef>* op_defs) { in GetRegisteredOps() argument
144 op_defs->push_back(p.second->op_def); in GetRegisteredOps()
|
| D | op.h | 97 void GetRegisteredOps(std::vector<OpDef>* op_defs);
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 574 std::vector<tensorflow::OpDef> op_defs; in TF_GetAllOpList() local
575 tensorflow::OpRegistry::Global()->GetRegisteredOps(&op_defs); in TF_GetAllOpList()
577 for (const auto& op : op_defs) { in TF_GetAllOpList()
|