MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Find

Method Find

tensorflow/core/kernels/matmul_op_fused.cc:145–154  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

143 public:
144 BlasPlansAutotuneCache() {}
145 bool Find(const se::BatchMatmulParameters& params,
146 se::blas::AlgorithmConfig* config) const {
147 absl::MutexLock lock(&mu_);
148 auto iter = blas_plans_algorithms_map_.find(params);
149 if (iter == blas_plans_algorithms_map_.end()) {
150 return false;
151 }
152 *config = iter->second;
153 return true;
154 }
155 void Insert(const se::BatchMatmulParameters& params,
156 const se::blas::AlgorithmConfig& config) {
157 absl::MutexLock lock(&mu_);

Callers 1

AutotuneMatmulFunction · 0.45

Calls 2

findMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected