Method Find

tensorflow/stream_executor/matmul_util.h:136–144

class BlasLtMatmulPlanMap {
 public:
  const blas::PlanAndAlgorithms* Find(
      const BatchMatmulParameters& params) const {
    absl::MutexLock lock(&mu_);
    auto iter = params_plan_map_.find(params);
    if (iter == params_plan_map_.end()) {
      return nullptr;
    }
    return &iter->second;
  }
  const blas::PlanAndAlgorithms* Insert(const BatchMatmulParameters& params,
                                        blas::PlanAndAlgorithms value) {
    absl::MutexLock lock(&mu_);