Searched refs:specialized_func (Results 1 – 1 of 1) sorted by relevance
351 const FunctionSpecialization& specialized_func) { in AddSpecializedFunction() argument352 specialized_functions_.emplace(sig, specialized_func); in AddSpecializedFunction()368 const FunctionSpecialization& specialized_func) { in AddTensorMapping() argument369 for (const auto& pair : specialized_func.output_mapping) { in AddTensorMapping()837 FunctionDef specialized_func; in SpecializeFunction() local838 TF_RETURN_IF_ERROR(MakeFunctionDef(item, flib, &specialized_func)); in SpecializeFunction()849 specialized_func.mutable_signature()->set_name(specialized_func_name); in SpecializeFunction()850 auto* specialized_attr = specialized_func.mutable_attr(); in SpecializeFunction()855 ctx->mutable_function_library()->AddFunctionDef(specialized_func)); in SpecializeFunction()