Home
last modified time | relevance | path

Searched refs:op_def_registry (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dinput_ops.py25 from tensorflow.python.framework import op_def_registry
71 return op.op_def or op_def_registry.get(op.type)
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dop_reg_gen.py29 from tensorflow.python.framework import op_def_registry
86 op_def = op_def_registry.get(op_name)
Dtfr_gen.py47 from tensorflow.python.framework import op_def_registry
269 op_def = op_def_registry.get(op_name)
696 return op_def_registry.get(op_name)
/external/tensorflow/tensorflow/python/framework/
Dgraph_to_function_def.py26 from tensorflow.python.framework import op_def_registry
65 return op.op_def or op_def_registry.get(op.type)
Dimporter.py29 from tensorflow.python.framework import op_def_registry
106 op_def = op_def_registry.get(node.op)
159 op_def = op_def_registry.get(node.op)
Dauto_control_deps.py27 from tensorflow.python.framework import op_def_registry
382 if (op_def_registry.get(op.type) is None or
Dmeta_graph.py40 from tensorflow.python.framework import op_def_registry
199 op_def = op_def_registry.get(op)
484 op_def = op_def_registry.get(node_def.op)
Dop_def_library.py31 from tensorflow.python.framework import op_def_registry
322 op_def = op_def_registry.get(op_type_name)
/external/tensorflow/tensorflow/python/eager/
Dgradient_input_output_exclusions.py40 from tensorflow.python.framework import op_def_registry
286 op_def = op_def_registry.get(op_type)
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_graphs.py21 from tensorflow.python.framework import op_def_registry
320 op_def = op_def_registry.get(node.op)
/external/tensorflow/tensorflow/python/compiler/xla/
Djit_test.py26 from tensorflow.python.framework import op_def_registry
40 op_def = op_def_registry.get(node_def.op)
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py31 from tensorflow.python.framework import op_def_registry
458 op_def = op_def_registry.get(node_def.op)
DBUILD554 "//tensorflow/python:op_def_registry",
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1845 auto* op_def_registry = OpRegistry::Global(); in Create() local
1849 auto status = op_def_registry->LookUpOpDef(n.op, &op_def); in Create()
1893 auto* op_def_registry = OpRegistry::Global(); in Define() local
1915 TF_CHECK_OK(op_def_registry->LookUpOpDef(n->op(), &op_def)) << n->op(); in Define()
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt59 [//tensorflow/core:framework_internal_impl] # op_def_registry, dtypes
/external/tensorflow/tensorflow/python/
DBUILD698 ":op_def_registry",
719 ":op_def_registry",
908 ":op_def_registry",
922 ":op_def_registry",
1018 ":op_def_registry",
1029 srcs = ["framework/op_def_registry.cc"],
1040 name = "op_def_registry",
1041 srcs = ["framework/op_def_registry.py"],
1420 ":op_def_registry",
5320 "//tensorflow/core:framework_internal_impl", # op_def_registry
/external/tensorflow/tensorflow/python/debug/
DBUILD159 "//tensorflow/python:op_def_registry",
178 "//tensorflow/python:op_def_registry",