| /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 | 197 op_defs = []
201 op_defs.append(op_def)
202 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/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/llvm-project/mlir/utils/spirv/ |
| D | gen_spirv_dialect.py | 932 op_defs = []
938 op_defs.append(
945 op_defs.append(name_op_map[opname])
948 op_defs = [header] + op_defs + [footer]
949 content = AUTOGEN_OP_DEF_SEPARATOR.join(op_defs)
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 343 def __init__(self, op_defs): argument
345 self._op_defs = op_defs
599 def __init__(self, ctx, op_defs): argument
603 self._op_defs = op_defs
1395 def __init__(self, op_defs): argument
1396 self._op_defs = op_defs
1416 def tfr_gen(func, op_defs): argument
1418 mlir_code, _ = TfrGen(op_defs).transform(func, None)
1425 op_defs = OpDefCache()
1462 mlir_funcs = [tfr_gen(func, op_defs) for func in py_funcs]
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 139 void OpRegistry::GetRegisteredOps(std::vector<OpDef>* op_defs) { in GetRegisteredOps() argument
143 op_defs->push_back(p.second->op_def); in GetRegisteredOps()
|
| D | op.h | 94 void GetRegisteredOps(std::vector<OpDef>* op_defs);
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2_test.py | 1082 for standard_def, op_defs in duplicates.items():
1084 op_defs = [i for i in op_defs if APPLY_SCOPE in i.name]
1087 if len(op_defs) < _NUM_LEARNERS:
1092 name, op_type = op_defs[0].name, op_defs[0].type
1097 num_duplicates += len(op_defs)
1099 for level in op_defs[0].traceback:
1104 op_defs[0].name,
|
| /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()
|