Home
last modified time | relevance | path

Searched refs:OpRegistry (Results 1 – 25 of 224) sorted by relevance

123456789

/external/tensorflow/tensorflow/core/framework/
Dop.cc46 OpRegistry::OpRegistry() : initialized_(false) {} in OpRegistry() function in tensorflow::OpRegistry
48 OpRegistry::~OpRegistry() { in ~OpRegistry()
52 void OpRegistry::Register(const OpRegistrationDataFactory& op_data_factory) { in Register()
61 Status OpRegistry::LookUp(const string& op_type_name, in LookUp()
76 Status OpRegistry::LookUpSlow(const string& op_type_name, in LookUpSlow()
125 void OpRegistry::GetRegisteredOps(std::vector<OpDef>* op_defs) { in GetRegisteredOps()
133 void OpRegistry::GetOpRegistrationData( in GetOpRegistrationData()
142 Status OpRegistry::SetWatcher(const Watcher& watcher) { in SetWatcher()
152 void OpRegistry::Export(bool include_internal, OpList* ops) const { in Export()
171 void OpRegistry::DeferRegistrations() { in DeferRegistrations()
[all …]
Dload_library.cc58 Status s = OpRegistry::Global()->ProcessRegistrations(); in LoadLibrary()
62 TF_RETURN_IF_ERROR(OpRegistry::Global()->SetWatcher( in LoadLibrary()
78 OpRegistry::Global()->DeferRegistrations(); in LoadLibrary()
81 s = OpRegistry::Global()->ProcessRegistrations(); in LoadLibrary()
84 OpRegistry::Global()->ClearDeferredRegistrations(); in LoadLibrary()
85 TF_RETURN_IF_ERROR(OpRegistry::Global()->SetWatcher(nullptr)); in LoadLibrary()
88 TF_RETURN_IF_ERROR(OpRegistry::Global()->SetWatcher(nullptr)); in LoadLibrary()
Dop_registration_test.cc27 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()
Dfunction_test.cc53 return OpRegistry::Global()->LookUpOpDef(op, sig); in GetOpSig()
972 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1013 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1048 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
1073 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1087 FunctionLibraryDefinition lib_def2(OpRegistry::Global(), proto); in TEST()
1098 FunctionLibraryDefinition lib_def3(OpRegistry::Global(), proto); in TEST()
1110 FunctionLibraryDefinition lib_def4(OpRegistry::Global(), proto); in TEST()
1133 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
[all …]
Dop.h65 class OpRegistry : public OpRegistryInterface {
69 OpRegistry();
70 ~OpRegistry() override;
87 static OpRegistry* Global();
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc318 Graph g(OpRegistry::Global()); in TEST_F()
337 Graph g(OpRegistry::Global()); in TEST_F()
355 Graph g(OpRegistry::Global()); in TEST_F()
372 Graph g(OpRegistry::Global()); in TEST_F()
433 Graph g(OpRegistry::Global()); in TEST_F()
448 Graph g(OpRegistry::Global()); in TEST_F()
463 Graph g(OpRegistry::Global()); in TEST_F()
478 Graph g(OpRegistry::Global()); in TEST_F()
493 Graph g(OpRegistry::Global()); in TEST_F()
508 Graph g(OpRegistry::Global()); in TEST_F()
[all …]
Dshape_refiner_test.cc84 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()
168 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
185 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
230 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
248 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
265 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
281 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
[all …]
Dconstant_folding_test.cc117 Graph g(OpRegistry::Global()); in TEST_F()
175 Graph g1(OpRegistry::Global()); in TEST_F()
177 Graph g2(OpRegistry::Global()); in TEST_F()
192 Graph g(OpRegistry::Global()); in TEST_F()
218 Graph g(OpRegistry::Global()); in TEST_F()
242 Graph g(OpRegistry::Global()); in TEST_F()
271 Graph g(OpRegistry::Global()); in TEST_F()
310 Graph g(OpRegistry::Global()); in TEST_F()
340 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST_F()
379 Graph g(OpRegistry::Global()); in TEST_F()
[all …]
Dexecutor_test.cc147 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
175 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
232 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
265 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
284 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
303 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
322 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
374 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
399 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
420 Graph* g = new Graph(OpRegistry::Global()); in BM_executor()
[all …]
Dmemory_types_test.cc26 Graph* g = new Graph(OpRegistry::Global()); in TEST()
44 Graph* g = new Graph(OpRegistry::Global()); in TEST()
72 Graph* g = new Graph(OpRegistry::Global()); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dvalidate_test.cc47 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()
/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops_test.cc62 Graph* g = new Graph(OpRegistry::Global()); in SGD()
68 Graph* g = new Graph(OpRegistry::Global()); in SGD()
90 Graph* g = new Graph(OpRegistry::Global()); in Adagrad()
99 Graph* g = new Graph(OpRegistry::Global()); in Adagrad()
123 Graph* g = new Graph(OpRegistry::Global()); in Momentum()
132 Graph* g = new Graph(OpRegistry::Global()); in Momentum()
157 Graph* g = new Graph(OpRegistry::Global()); in Adam()
168 Graph* g = new Graph(OpRegistry::Global()); in Adam()
200 Graph* g = new Graph(OpRegistry::Global()); in RMSProp()
211 Graph* g = new Graph(OpRegistry::Global()); in RMSProp()
[all …]
Drandom_op_test.cc41 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()
89 Graph* g = new Graph(OpRegistry::Global()); in BM_cpu_RandomGamma()
Dscan_ops_test.cc26 auto* g = new Graph(OpRegistry::Global()); in LargeOneDCumsum()
37 auto* g = new Graph(OpRegistry::Global()); in ColCumsum()
48 auto* g = new Graph(OpRegistry::Global()); in RowCumsum()
59 auto* g = new Graph(OpRegistry::Global()); in ThreeDYCumsum()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc89 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
115 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
136 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
158 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
178 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
213 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
245 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
313 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
369 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
381 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
[all …]
Dpartially_decluster_pass_test.cc96 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in PartiallyDecluster()
128 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
164 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
193 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
223 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
254 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
315 auto graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
343 std::unique_ptr<Graph> graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
368 std::unique_ptr<Graph> graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
392 std::unique_ptr<Graph> graph = absl::make_unique<Graph>(OpRegistry::Global()); in TEST()
[all …]
Dextract_outside_compilation_pass_test.cc52 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()
276 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
288 FunctionLibraryDefinition fld(OpRegistry::Global(), fdl); in TEST_F()
410 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
416 FunctionLibraryDefinition fld(OpRegistry::Global(), fdl); in TEST_F()
456 std::unique_ptr<Graph> g(new Graph(OpRegistry::Global())); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils_test.cc116 FunctionLibraryDefinition flib_0(OpRegistry::Global(), fdef_base); in TEST()
119 FunctionLibraryDefinition flib_1(OpRegistry::Global(), fdef_base); in TEST()
120 FunctionLibraryDefinition flib_to_add(OpRegistry::Global(), fdef_to_add); in TEST()
150 FunctionLibraryDefinition flib_0(OpRegistry::Global(), fdef_base); in TEST()
158 FunctionLibraryDefinition flib_1(OpRegistry::Global(), fdef_base); in TEST()
159 FunctionLibraryDefinition flib_to_add(OpRegistry::Global(), fdef_to_add); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_stream_util_test.cc38 Graph g(OpRegistry::Global()); in TEST_F()
58 Graph g(OpRegistry::Global()); in TEST_F()
69 Graph g(OpRegistry::Global()); in TEST_F()
88 Graph g(OpRegistry::Global()); in TEST_F()
113 Graph g(OpRegistry::Global()); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc60 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in SetUp()
176 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
193 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
243 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
296 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
347 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
390 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
430 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
471 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST_F()
485 std::unique_ptr<Graph> graph_copy(new Graph(OpRegistry::Global())); in TEST_F()
[all …]
Dtf2xla_util_test.cc223 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in TEST()
291 FunctionLibraryDefinition fld(OpRegistry::Global(), proto); in TEST()
335 FunctionLibraryDefinition fld(OpRegistry::Global(), {}); in TEST()
342 Graph graph(OpRegistry::Global()); in TEST()
360 Graph graph(OpRegistry::Global()); in TEST()
367 FunctionLibraryDefinition fld(OpRegistry::Global(), {}); in TEST()
375 Graph graph(OpRegistry::Global()); in TEST()
394 Graph graph(OpRegistry::Global()); in TEST()
Dconst_analysis_test.cc80 Graph graph(OpRegistry::Global()); in TEST()
102 Graph graph(OpRegistry::Global()); in TEST()
128 Graph graph(OpRegistry::Global()); in TEST()
153 Graph graph(OpRegistry::Global()); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc181 Graph graph_1(OpRegistry::Global()); in TEST_F()
192 Graph graph_2(OpRegistry::Global()); in TEST_F()
216 Graph graph_1(OpRegistry::Global()); in TEST_F()
235 Graph graph_1(OpRegistry::Global()); in TEST_F()
254 Graph graph_1(OpRegistry::Global()); in TEST_F()
294 Graph graph_0(OpRegistry::Global()); in TEST_F()
305 Graph graph_1(OpRegistry::Global()); in TEST_F()
366 Graph graph(OpRegistry::Global()); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc244 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
326 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
381 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
480 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
529 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
594 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
633 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
660 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
701 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
741 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dshape_function_test.cc25 OpRegistry* op_registry = OpRegistry::Global(); in TEST()

123456789