Searched refs:overlay_lib_ (Results 1 – 2 of 2) sorted by relevance
188 overlay_lib_(overlay_lib) {} in MultiDeviceFunctionData()197 FunctionLibraryDefinition overlay_lib_; member
588 optimization_options.flib_def = &data->overlay_lib_; in InstantiateMultiDevice()629 &data->overlay_lib_, device_set, cpu_device, &graph); in InstantiateMultiDevice()663 FunctionNameGenerator name_generator(&data->overlay_lib_, function_name); in InstantiateMultiDevice()679 TF_RETURN_IF_ERROR(data->overlay_lib_.AddFunctionDef(shard)); in InstantiateMultiDevice()683 opts.overlay_lib = &data->overlay_lib_; in InstantiateMultiDevice()