tensorflow_model_analysis/evaluators/query_metrics/ndcg.py
tensorflow_model_analysis/metrics/ndcg.py

tensorflow_model_analysis/evaluators/keras_util.py
tensorflow_model_analysis/metrics/tf_metric_wrapper.py