Searched refs:fdef_lib (Results 1 – 12 of 12) sorted by relevance
58 Status BuildXlaOps(const Scope& s, const FunctionDefLibrary& fdef_lib, in BuildXlaOps() argument62 FunctionLibraryDefinition flib_def(graph->op_registry(), fdef_lib); in BuildXlaOps()123 FunctionDefLibrary fdef_lib; in CreateFunctionDefLibWithConstFunction() local129 *fdef_lib.add_function() = std::move(func); in CreateFunctionDefLibWithConstFunction()130 return fdef_lib; in CreateFunctionDefLibWithConstFunction()134 FunctionDefLibrary fdef_lib; in CreateFunctionDefLibWithInt32Input() local140 *fdef_lib.add_function() = std::move(func); in CreateFunctionDefLibWithInt32Input()141 return fdef_lib; in CreateFunctionDefLibWithInt32Input()148 FunctionDefLibrary fdef_lib = in TEST_F() local150 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()[all …]
206 void AddRefFunctionFunctionDef(FunctionDefLibrary* fdef_lib) { in AddRefFunctionFunctionDef() argument214 *fdef_lib->add_function() = make_ref_float; in AddRefFunctionFunctionDef()217 void AddRegularFunctionFunctionDef(FunctionDefLibrary* fdef_lib) { in AddRegularFunctionFunctionDef() argument223 *fdef_lib->add_function() = make_regular_float; in AddRegularFunctionFunctionDef()229 FunctionDefLibrary fdef_lib; in TEST() local234 AddRefFunctionFunctionDef(&fdef_lib); in TEST()237 AddRegularFunctionFunctionDef(&fdef_lib); in TEST()240 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST()
247 tensorflow::FunctionDefLibrary fdef_lib; in TF_GraphCopyFunction() local248 *fdef_lib.add_function() = func->fdef; in TF_GraphCopyFunction()250 *fdef_lib.add_function() = grad->fdef; in TF_GraphCopyFunction()251 tensorflow::GradientDef* gdef = fdef_lib.add_gradient(); in TF_GraphCopyFunction()257 status->status = g->graph.AddFunctionLibrary(fdef_lib); in TF_GraphCopyFunction()
196 const auto& fdef_lib = gdef.library(); in CreateFunctionsFromTextProto() local197 if (fdef_lib.gradient_size() > 0) { in CreateFunctionsFromTextProto()204 for (const FunctionDef& fdef : fdef_lib.function()) { in CreateFunctionsFromTextProto()
168 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() argument169 return graph_.AddFunctionLibrary(fdef_lib); in AddFunctionLibrary()
462 FunctionDefLibrary fdef_lib; in TEST_F() local463 *fdef_lib.add_function() = test::function::XTimesTwo(); in TEST_F()464 *fdef_lib.add_function() = test::function::XTimesFour(); in TEST_F()465 TF_ASSERT_OK(in_.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
631 Status Graph::AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() argument633 if (fdef_lib.function_size() > 0 && versions_->min_consumer() < 12) { in AddFunctionLibrary()636 return ops_.AddLibrary(fdef_lib); in AddFunctionLibrary()
558 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib);
144 FunctionDefLibrary fdef_lib; in TEST() local145 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST()
623 FunctionDefLibrary fdef_lib; in TEST_F() local624 *(fdef_lib.add_function()) = fdef; in TEST_F()628 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()663 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST_F()1562 FunctionDefLibrary fdef_lib; in TEST_F() local1563 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST_F()1635 FunctionDefLibrary fdef_lib; in TEST_F() local1636 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST_F()
341 FunctionDefLibrary fdef_lib; in AddNoinlineFunctionToGraph() local342 *(fdef_lib.add_function()) = GetNoinlineFunctionDef(); in AddNoinlineFunctionToGraph()343 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(fdef_lib)); in AddNoinlineFunctionToGraph()
97 const FunctionDefLibrary& fdef_lib) { in GetFunctionDefLibraryStub() argument99 for (const FunctionDef& fn : fdef_lib.function()) { in GetFunctionDefLibraryStub()106 *stub.mutable_gradient() = fdef_lib.gradient(); in GetFunctionDefLibraryStub()