Searched refs:ComputeMean (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.h | 35 …virtual void ComputeMean(const std::vector<std::vector<size_t>> &shapes, size_t n, size_t server_n… in ComputeMean() function
66 void ComputeMean(const std::vector<std::vector<size_t>> &shapes, size_t n, size_t server_num,
77 void ComputeMean(const std::vector<std::vector<size_t>> &shapes, size_t n, size_t server_num,
|
| D | optimizer_info.cc | 101 void DenseOptimInfo::ComputeMean(const std::vector<std::vector<size_t>> &, size_t n, size_t, size_t… in ComputeMean() function in mindspore::ps::DenseOptimInfo
182 void SparseOptimInfo::ComputeMean(const std::vector<std::vector<size_t>> &shapes, size_t n, size_t … in ComputeMean() function in mindspore::ps::SparseOptimInfo
|
| D | parameter_server.cc | 310 optim_info->ComputeMean(shapes, worker_num_, pserver_num_, server_node_->rank_id()); in UpdateWeights()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | instance_norm_impl.cuh | 25 void ComputeMean(const size_t N, const size_t C, float *dgamma, float *dbeta, const float *ws_dgamm…
|
| D | instance_norm_impl.cu | 82 void ComputeMean(const size_t N, const size_t C, in ComputeMean() function
|
| /third_party/skia/third_party/externals/angle2/src/tests/perf_tests/ |
| D | ANGLEPerfTest.cpp | 218 double ComputeMean(const std::vector<double> &values) in ComputeMean() function
315 double mean = ComputeMean(mTestTrialResults); in run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | instance_norm_grad_gpu_kernel.h | 99 … ComputeMean(N, C, dgamma, dbeta, ws_dgamma, ws_dbeta, reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|