| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 52 OpRegistry::OpRegistry() in OpRegistry() function in tensorflow::OpRegistry
55 OpRegistry::~OpRegistry() { in ~OpRegistry()
59 void OpRegistry::Register(const OpRegistrationDataFactory& op_data_factory) { in Register()
84 Status OpRegistry::LookUp(const string& op_type_name, in LookUp()
90 const OpRegistrationData* OpRegistry::LookUp(const string& op_type_name) const { in LookUp()
103 const OpRegistrationData* OpRegistry::LookUpSlow( in LookUpSlow()
139 void OpRegistry::GetRegisteredOps(std::vector<OpDef>* op_defs) { in GetRegisteredOps()
147 void OpRegistry::GetOpRegistrationData( in GetOpRegistrationData()
156 Status OpRegistry::SetWatcher(const Watcher& watcher) { in SetWatcher()
166 void OpRegistry::Export(bool include_internal, OpList* ops) const { in Export()
[all …]
|
| D | load_library.cc | 58 Status s = OpRegistry::Global()->ProcessRegistrations(); in LoadDynamicLibrary()
62 TF_RETURN_IF_ERROR(OpRegistry::Global()->SetWatcher( in LoadDynamicLibrary()
78 OpRegistry::Global()->DeferRegistrations(); in LoadDynamicLibrary()
81 s = OpRegistry::Global()->ProcessRegistrations(); in LoadDynamicLibrary()
84 OpRegistry::Global()->ClearDeferredRegistrations(); in LoadDynamicLibrary()
85 TF_RETURN_IF_ERROR(OpRegistry::Global()->SetWatcher(nullptr)); in LoadDynamicLibrary()
88 TF_RETURN_IF_ERROR(OpRegistry::Global()->SetWatcher(nullptr)); in LoadDynamicLibrary()
|
| D | op_registration_test.cc | 27 void Register(const string& op_name, OpRegistry* registry) { in Register()
37 std::unique_ptr<OpRegistry> registry(new OpRegistry); in TEST()
46 std::unique_ptr<OpRegistry> registry(new OpRegistry); in TEST()
|
| D | function_test.cc | 56 return OpRegistry::Global()->LookUpOpDef(op, sig); in GetOpSig()
973 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
981 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1011 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1036 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
1058 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1072 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1085 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1099 FunctionLibraryDefinition lib_def2(OpRegistry::Global(), proto); in TEST()
[all …]
|
| D | op.h | 66 class OpRegistry : public OpRegistryInterface {
70 OpRegistry();
71 ~OpRegistry() override;
91 static OpRegistry* Global();
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils_test.cc | 27 using tensorflow::OpRegistry;
32 void Register(const std::string& op_name, OpRegistry* registry) { in Register()
42 std::unique_ptr<OpRegistry> registry(new OpRegistry); in TEST()
48 std::unique_ptr<OpRegistry> registry(new OpRegistry); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_test.cc | 406 Graph g(OpRegistry::Global()); in TEST_F()
425 Graph g(OpRegistry::Global()); in TEST_F()
445 Graph g(OpRegistry::Global()); in TEST_F()
468 Graph g(OpRegistry::Global()); in TEST_F()
485 Graph g(OpRegistry::Global()); in TEST_F()
548 Graph g(OpRegistry::Global()); in TEST_F()
563 Graph g(OpRegistry::Global()); in TEST_F()
578 Graph g(OpRegistry::Global()); in TEST_F()
593 Graph g(OpRegistry::Global()); in TEST_F()
608 Graph g(OpRegistry::Global()); in TEST_F()
[all …]
|
| D | shape_refiner_test.cc | 84 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TestStridedSlice()
129 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
136 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
153 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
170 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
215 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
233 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
250 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
266 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
302 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
[all …]
|
| D | memory_types_test.cc | 26 Graph* g = new Graph(OpRegistry::Global()); in TEST()
41 Graph* g = new Graph(OpRegistry::Global()); in TEST()
60 Graph* g = new Graph(OpRegistry::Global()); in TEST()
|
| D | replicate_per_replica_nodes_test.cc | 93 Graph graph(OpRegistry::Global()); in TEST()
138 Graph graph(OpRegistry::Global()); in TEST()
180 Graph graph(OpRegistry::Global()); in TEST()
230 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST()
236 Graph graph(OpRegistry::Global()); in TEST()
260 Graph graph(OpRegistry::Global()); in TEST()
307 Graph graph(OpRegistry::Global()); in TEST()
|
| D | executor_test.cc | 159 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
187 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
244 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
277 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
296 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
315 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
334 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
386 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
411 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
427 auto g = absl::make_unique<Graph>(OpRegistry::Global()); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | validate_test.cc | 47 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global())); in TEST()
62 Status s = graph::ValidateGraphDef(graph_def, *OpRegistry::Global()); in TEST()
67 TF_ASSERT_OK(AddDefaultAttrsToGraphDef(&graph_def, *OpRegistry::Global(), 0)); in TEST()
70 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global())); in TEST()
85 Status s = graph::ValidateGraphDef(graph_def, *OpRegistry::Global()); in TEST()
90 TF_ASSERT_OK(AddDefaultAttrsToGraphDef(&graph_def, *OpRegistry::Global(), 0)); in TEST()
93 s = graph::ValidateGraphDef(graph_def, *OpRegistry::Global()); in TEST()
|
| D | validate.h | 52 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_test.cc | 91 Graph* g = new Graph(OpRegistry::Global()); in SGD()
97 Graph* g = new Graph(OpRegistry::Global()); in SGD()
123 Graph* g = new Graph(OpRegistry::Global()); in Adagrad()
132 Graph* g = new Graph(OpRegistry::Global()); in Adagrad()
159 Graph* g = new Graph(OpRegistry::Global()); in SparseAdagrad()
168 Graph* g = new Graph(OpRegistry::Global()); in SparseAdagrad()
204 Graph* g = new Graph(OpRegistry::Global()); in Momentum()
213 Graph* g = new Graph(OpRegistry::Global()); in Momentum()
242 Graph* g = new Graph(OpRegistry::Global()); in Adam()
253 Graph* g = new Graph(OpRegistry::Global()); in Adam()
[all …]
|
| D | random_op_test.cc | 41 Graph* g = new Graph(OpRegistry::Global()); in RandomUniform()
48 Graph* g = new Graph(OpRegistry::Global()); in RandomNormal()
55 Graph* g = new Graph(OpRegistry::Global()); in TruncatedNormal()
94 Graph* g = new Graph(OpRegistry::Global()); in BM_cpu_RandomGamma()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test.cc | 94 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
119 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
139 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
160 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
179 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
211 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
252 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
289 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
380 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
435 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
[all …]
|
| D | rearrange_function_argument_pass_test.cc | 52 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
67 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
80 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
85 FunctionLibraryDefinition fld(OpRegistry::Global(), fdl); in TEST()
109 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
185 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
200 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
205 FunctionLibraryDefinition fld(OpRegistry::Global(), fdl); in TEST()
220 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
|
| D | extract_outside_compilation_pass_test.cc | 52 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
128 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
154 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
178 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
204 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST()
277 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
289 FunctionLibraryDefinition fld(OpRegistry::Global(), fdl); in TEST_F()
401 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
407 FunctionLibraryDefinition fld(OpRegistry::Global(), fdl); in TEST_F()
437 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
[all …]
|
| D | partially_decluster_pass_test.cc | 125 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
161 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
190 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
220 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
251 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
312 auto graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
340 std::unique_ptr<Graph> graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
365 std::unique_ptr<Graph> graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
389 std::unique_ptr<Graph> graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
413 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_test.cc | 47 OpRegistry::Global()->Export(false, &ops); in TEST()
121 OpRegistry::Global()->Export(false, &op_defs); in TEST()
183 OpRegistry::Global()->Export(false, &op_defs); in TEST()
242 OpRegistry::Global()->Export(false, &op_defs); in TEST()
303 OpRegistry::Global()->Export(false, &op_defs); in TEST()
362 OpRegistry::Global()->Export(false, &op_defs); in TEST()
419 OpRegistry::Global()->Export(false, &op_defs); in TEST()
468 OpRegistry::Global()->Export(false, &op_defs); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment_test.cc | 128 Graph g(OpRegistry::Global()); in TEST_F()
152 Graph g(OpRegistry::Global()); in TEST_F()
205 Graph g(OpRegistry::Global()); in TEST_F()
213 Graph g(OpRegistry::Global()); in TEST_F()
229 Graph g(OpRegistry::Global()); in TEST_F()
240 Graph g(OpRegistry::Global()); in TEST_F()
263 Graph g(OpRegistry::Global()); in TEST_F()
295 Graph g(OpRegistry::Global()); in TEST_F()
343 Graph g(OpRegistry::Global()); in TEST_F()
361 Graph g(OpRegistry::Global()); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 225 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
296 FunctionLibraryDefinition fld(OpRegistry::Global(), proto); in TEST()
340 FunctionLibraryDefinition fld(OpRegistry::Global(), {}); in TEST()
347 Graph graph(OpRegistry::Global()); in TEST()
365 Graph graph(OpRegistry::Global()); in TEST()
372 FunctionLibraryDefinition fld(OpRegistry::Global(), {}); in TEST()
380 Graph graph(OpRegistry::Global()); in TEST()
399 Graph graph(OpRegistry::Global()); in TEST()
425 FunctionLibraryDefinition fld(OpRegistry::Global(), {}); in TEST()
433 Graph graph(OpRegistry::Global()); in TEST()
[all …]
|
| D | xla_compiler_test.cc | 68 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in SetUp()
184 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
201 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
250 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
294 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
335 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
387 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
440 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
472 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
503 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
[all …]
|
| D | resource_util_test.cc | 85 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
120 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
160 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
203 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
252 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
315 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
392 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST()
449 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST()
499 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | shape_function_test.cc | 25 OpRegistry* op_registry = OpRegistry::Global(); in TEST()
|