Searched refs:function_library_ (Results 1 – 4 of 4) sorted by relevance
165 function_library_ = lib; in set_function_library_for_shape_inference()169 return function_library_ != nullptr; in function_shape_inference_supported()290 const tensorflow::FunctionLibraryDefinition* function_library_ = nullptr; variable
144 return this->function_library_->LookUpOpDef(op, sig); in InferShapesForFunction()148 Graph* new_graph = new Graph(function_library_); in InferShapesForFunction()639 if (function_library_ && op_reg_data->is_function_op) { in RunShapeFn()642 auto* func_def = function_library_->Find(op_reg_data->op_def.name()); in RunShapeFn()
278 function_library_(OpRegistry::Global(), graph.library()), in FunctionOptimizerContext()289 return function_library_; in function_library()293 return &function_library_; in mutable_function_library()419 &function_library_, optimizer_opts)); in InitializeFunctionLibraryRuntime()428 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::__anon35816ea30111::FunctionOptimizerContext
553 function_library_(OpRegistry::Global(), graph.graph()->library()), in SymbolicShapeRefiner()1042 CHECK_NOTNULL(function_library_.Find(function_node->op())); in AddFunction()1046 MakeGrapplerFunctionItem(*function_def, function_library_, in AddFunction()1072 TF_RETURN_IF_ERROR(function_library_.LookUp(node->op(), &node_ctx.op_data)); in AddNode()1696 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::SymbolicShapeRefiner