Searched refs:GetFunctionLibraryDefinition (Results 1 – 14 of 14) sorted by relevance
166 if (IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)) { in Compile()208 if (flib.GetFunctionLibraryDefinition()->Find(node.def().op())) { in GetFunctionNameAndAttr()221 TF_RET_CHECK(IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)); in CompileFunctionalNode()
1057 flib_runtime_->GetFunctionLibraryDefinition()); in CompileGraph()
71 flr.GetFunctionLibraryDefinition()->Find(node_def.name()); in CompilationRequested()92 Status status = flr.GetFunctionLibraryDefinition()->GetAttr( in CompilationRequested()
187 options.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in Compile()
37 params.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in Compute()
38 params.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in Optimize()
217 *ctx->function_library()->GetFunctionLibraryDefinition()); in SetIteratorFromDataset()341 params.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in InitializeFromIterator()488 *ctx->function_library()->GetFunctionLibraryDefinition()); in CreatePrivateFLR()
107 lib->GetFunctionLibraryDefinition()->Find(node_op) == nullptr && in ShouldOwnKernel()
676 virtual const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
89 ->GetFunctionLibraryDefinition() in Init()92 function_def = flr_->GetFunctionLibraryDefinition()->Find(ndef.op()); in Init()
192 const FunctionLibraryDefinition* GetFunctionLibraryDefinition()282 FunctionLibraryRuntimeOverlay::GetFunctionLibraryDefinition() const { in GetFunctionLibraryDefinition() function in tensorflow::FunctionLibraryRuntimeOverlay284 : base_flr_->GetFunctionLibraryDefinition(); in GetFunctionLibraryDefinition()338 const FunctionLibraryDefinition* GetFunctionLibraryDefinition() in GetFunctionLibraryDefinition() function in tensorflow::FunctionLibraryRuntimeImpl1795 const FunctionLibraryDefinition* fld = lib->GetFunctionLibraryDefinition(); in ExpandInlineFunctions()1799 if (!IsFunctionCall(*lib->GetFunctionLibraryDefinition(), *node)) { in ExpandInlineFunctions()
131 const FunctionLibraryDefinition* flib = lib.GetFunctionLibraryDefinition(); in FillOutputDevices()
289 options.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in CompileToLocalExecutable()
167 auto fdef = lib->GetFunctionLibraryDefinition()->Find(funcdef_name_); in ConstructFunctionHandle()