Searched refs:specialized_func (Results 1 – 1 of 1) sorted by relevance
310 const FunctionSpecialization& specialized_func) { in AddSpecializedFunction() argument311 specialized_functions_.emplace(sig, specialized_func); in AddSpecializedFunction()327 const FunctionSpecialization& specialized_func) { in AddTensorMapping() argument328 for (const auto& pair : specialized_func.output_mapping) { in AddTensorMapping()744 FunctionDef specialized_func; in SpecializeFunction() local745 TF_RETURN_IF_ERROR(MakeFunctionDef(item, flib, &specialized_func)); in SpecializeFunction()756 specialized_func.mutable_signature()->set_name(specialized_func_name); in SpecializeFunction()757 auto* specialized_attr = specialized_func.mutable_attr(); in SpecializeFunction()761 TF_RETURN_IF_ERROR(ctx->function_library().AddFunctionDef(specialized_func)); in SpecializeFunction()