| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | lstm_grad_weight_gpu_kernel.h | 42 weight_size_(0), in LstmGradWeightGpuKernel()
159 … cudnnGetRNNParamsSize(handle_, rnn_desc_, x_desc_[0], &weight_size_, cudnn_data_type_), in Init()
161 if (weight_size != weight_size_) { in Init()
162 … MS_LOG(EXCEPTION) << "weight size: " << weight_size << " error, expect: " << weight_size_ << " ."; in Init()
164 int w_dims[3] = {SizeToInt(weight_size_ / sizeof(T)), 1, 1}; in Init()
200 output_size_list_.push_back(weight_size_); in InitSizeLists()
256 size_t weight_size_; variable
|
| D | lstm_gpu_kernel.h | 43 weight_size_(0), in LstmGpuKernel()
170 … cudnnGetRNNParamsSize(handle_, rnn_desc_, x_desc_[0], &weight_size_, cudnn_data_type_), in Init()
172 if (weight_size != weight_size_) { in Init()
173 … MS_LOG(EXCEPTION) << "weight size: " << weight_size << " error, expect: " << weight_size_ << " ."; in Init()
175 int w_dims[3] = {SizeToInt(weight_size_ / sizeof(T)), 1, 1}; in Init()
242 input_size_list_.push_back(weight_size_); in InitSizeLists()
274 size_t weight_size_; variable
|
| D | bce_with_logits_loss_kernel.h | 111 weight_size_ = 1; in Init()
113 weight_size_ *= weight_shape_[i]; in Init()
128 weight_size_ = 1; in ResetResource()
145 input_size_list_.push_back(weight_size_ * sizeof(T)); in InitSizeLists()
174 size_t weight_size_; variable
|
| D | lstm_grad_data_gpu_kernel.h | 43 weight_size_(0), in LstmGradDataGpuKernel()
184 … cudnnGetRNNParamsSize(handle_, rnn_desc_, dx_desc_[0], &weight_size_, cudnn_data_type_), in Init()
186 if (weight_size != weight_size_) { in Init()
187 … MS_LOG(EXCEPTION) << "weight size: " << weight_size << " error, expect: " << weight_size_ << " ."; in Init()
189 int w_dims[3] = {SizeToInt(weight_size_ / sizeof(T)), 1, 1}; in Init()
238 input_size_list_.push_back(weight_size_); in InitSizeLists()
306 size_t weight_size_; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | davinci_model.h | 37 weight_size_(weight_size), in task_info_list_()
49 uint64_t GetWeightSize() const { return weight_size_; } in GetWeightSize()
74 uint64_t weight_size_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | lstm_grad_cpu_kernel.cc | 134 weight_size_ += gate_size * (i == 0 ? input_size_ : hidden_size_ * num_directions_); in CheckParam()
137 weight_size_ = weight_size_ * num_directions_; in CheckParam()
189 user_weights_h_memory.set_data_handle(reinterpret_cast<float *>(inputs[3]->addr) + weight_size_); in Launch()
193 …bias_memory.set_data_handle(reinterpret_cast<float *>(inputs[3]->addr) + weight_size_ + weight_h_s… in Launch()
207 …_diff_weights_h_memory.set_data_handle(reinterpret_cast<float *>(outputs[3]->addr) + weight_size_); in Launch()
213 …diff_bias_memory.set_data_handle(reinterpret_cast<float *>(outputs[3]->addr) + weight_size_ + weig… in Launch()
|
| D | lstm_cpu_kernel.cc | 139 weight_size_ += gate_size * (i == 0 ? input_size_ : hidden_size_ * num_directions_); in CheckParam()
142 weight_size_ = weight_size_ * num_directions_; in CheckParam()
162 user_weights_h_memory.set_data_handle(reinterpret_cast<float *>(inputs[3]->addr) + weight_size_); in Launch()
167 …bias_memory.set_data_handle(reinterpret_cast<float *>(inputs[3]->addr) + weight_size_ + weight_h_s… in Launch()
|
| D | lstm_cpu_kernel.h | 41 int weight_size_{0};
|
| D | lstm_grad_cpu_kernel.h | 55 int64_t weight_size_{0};
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | get_model_kernel.h | 126 size_t weight_size_ = in Init() local
128 input_size_list_.push_back(weight_size_); in Init()
|
| D | update_model_kernel.h | 109 size_t weight_size_ = in Init() local
111 input_size_list_.push_back(weight_size_); in Init()
|
| D | fused_push_weight_kernel.h | 118 …size_t weight_size_ = std::accumulate(weight_shape.begin(), weight_shape.end(), sizeof(T), std::mu… in Init() local
119 input_size_list_.push_back(weight_size_); in Init()
|
| D | fused_pull_weight_kernel.h | 131 …size_t weight_size_ = std::accumulate(weight_shape.begin(), weight_shape.end(), sizeof(T), std::mu… in Init() local
132 input_size_list_.push_back(weight_size_); in Init()
|