| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/ |
| D | dataset_iterator_kernel.cc | 51 …std::vector<size_t> &DatasetIteratorKernel::GetOutputSizeList() const { return output_size_list_; } in GetOutputSizeList()
69 output_size_list_.push_back(bytes); in Init()
73 handle_ = GpuBufferMgr::GetInstance().Open(0, queue_name_, output_size_list_); in Init()
139 handle_ = GpuBufferMgr::GetInstance().Open(0, queue_name_, output_size_list_); in Launch()
155 for (size_t i = 0; i < output_size_list_.size(); i++) { in Launch()
158 … cudaMemcpyAsync(output_addr, addr, output_size_list_[i], cudaMemcpyDeviceToDevice, in Launch()
161 addr = reinterpret_cast<unsigned char *>(addr) + output_size_list_[i]; in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/common/ |
| D | atomcrdtocv_kernel.h | 58 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
84 output_size_list_.push_back(number * sizeof(T)); in InitSizeLists()
85 output_size_list_.push_back(number * sizeof(T)); in InitSizeLists()
87 output_size_list_.push_back(3 * atom_numbers * sizeof(T)); in InitSizeLists()
88 output_size_list_.push_back(3 * atom_numbers * sizeof(T1)); in InitSizeLists()
97 std::vector<size_t> output_size_list_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | relu_v2_gpu_kernel.h | 36 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
70 output_size_list_.clear(); in ResetResource()
78 output_size_list_.push_back(size); in InitSizeLists()
80 output_size_list_.push_back(mask_size); in InitSizeLists()
88 std::vector<size_t> output_size_list_; variable
|
| D | fused_add_relu_v2_gpu_kernel.h | 36 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
71 output_size_list_.clear(); in ResetResource()
80 output_size_list_.push_back(size); in InitSizeLists()
83 output_size_list_.push_back(size); in InitSizeLists()
91 std::vector<size_t> output_size_list_; variable
|
| D | nll_loss_gpu_kernel.h | 35 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
106 output_size_list_.clear(); in ResetResource()
117 output_size_list_.push_back(n_ * sizeof(T)); // loss output of shape (N,) in InitSizeLists()
119 output_size_list_.push_back(sizeof(T)); // scalar loss output in InitSizeLists()
121 output_size_list_.push_back(sizeof(S)); // total weight in InitSizeLists()
135 std::vector<size_t> output_size_list_; variable
|
| D | layer_norm_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
100 output_size_list_.push_back(input_row_ * input_col_ * sizeof(T)); in InitSizeLists()
101 output_size_list_.push_back(input_row_ * sizeof(T)); in InitSizeLists()
102 output_size_list_.push_back(input_row_ * sizeof(T)); in InitSizeLists()
107 std::vector<size_t> output_size_list_; variable
|
| D | dropout_gpu_kernel.h | 35 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
110 output_size_list_.clear(); in ResetResource()
120 …output_size_list_.push_back(input_size); // output size: the same with input s… in InitSizeLists()
121 …output_size_list_.push_back(input_size); // mask size: the same with input size in InitSizeLists()
134 std::vector<size_t> output_size_list_; variable
|
| D | flatten_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
71 output_size_list_.clear(); in ResetResource()
78 output_size_list_.push_back(input_size_); in InitSizeLists()
83 std::vector<size_t> output_size_list_; variable
|
| D | prelu_grad_gpu_kernel.h | 36 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
108 output_size_list_.clear(); in ResetResource()
118 output_size_list_.push_back(input_length_ * data_size); in InitSizeLists()
119 output_size_list_.push_back(weight_length_ * data_size); in InitSizeLists()
130 std::vector<size_t> output_size_list_; variable
|
| D | layer_norm_grad_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
104 output_size_list_.push_back(input_row_ * input_col_ * sizeof(T)); in InitSizeLists()
105 output_size_list_.push_back(param_dim_ * sizeof(T)); in InitSizeLists()
106 output_size_list_.push_back(param_dim_ * sizeof(T)); in InitSizeLists()
111 std::vector<size_t> output_size_list_; variable
|
| D | relu_grad_v2_gpu_kernel.h | 36 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
70 output_size_list_.clear(); in ResetResource()
78 output_size_list_.push_back(size); in InitSizeLists()
89 std::vector<size_t> output_size_list_; variable
|
| D | batch_norm_gpu_kernel.h | 38 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
168 output_size_list_.clear(); in ResetResource()
238 output_size_list_.push_back(output_size_); // output in InitSizeLists()
239 output_size_list_.push_back(para_size_); // save scale in InitSizeLists()
240 output_size_list_.push_back(reserve_size_); // reserve space in InitSizeLists()
241 output_size_list_.push_back(para_size_); // save mean in InitSizeLists()
242 output_size_list_.push_back(para_size_); // save variance in InitSizeLists()
319 std::vector<size_t> output_size_list_; variable
|
| D | sparse_ftrl_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
135 output_size_list_.push_back(variable_size_); in InitSizeLists()
136 output_size_list_.push_back(accumulation_size_); in InitSizeLists()
137 output_size_list_.push_back(linear_size_); in InitSizeLists()
155 output_size_list_.clear(); in ResetResource()
175 std::vector<size_t> output_size_list_; variable
|
| D | dropout3d_gpu_kernel.h | 35 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
128 output_size_list_.clear(); in ResetResource()
137 output_size_list_.push_back(input_size); // output size: the same as input size in InitSizeLists()
138 output_size_list_.push_back(mask_size); in InitSizeLists()
156 std::vector<size_t> output_size_list_; variable
|
| D | kl_div_loss_grad_kernel.h | 35 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
77 output_size_list_.push_back(input_size_ * sizeof(T)); in InitSizeLists()
78 output_size_list_.push_back(input_size_ * sizeof(T)); in InitSizeLists()
92 std::vector<size_t> output_size_list_; variable
|
| D | hsigmoid_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
78 output_size_list_.clear(); in ResetResource()
85 output_size_list_.push_back(input_size_ * sizeof(T)); in InitSizeLists()
92 std::vector<size_t> output_size_list_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | batchnorm_fold2_grad_gpu_kernel.h | 43 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
146 output_size_list_.push_back(weight_size); // d_batch_std in InitSizeLists()
147 output_size_list_.push_back(weight_size); // d_batch_mean in InitSizeLists()
148 output_size_list_.push_back(weight_size); // d_beta in InitSizeLists()
149 output_size_list_.push_back(weight_size); // d_gamma in InitSizeLists()
150 output_size_list_.push_back(input_size); // d_x in InitSizeLists()
169 std::vector<size_t> output_size_list_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/restrain/ |
| D | restrain_force_atom_energy_virial_kernel.h | 60 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
87 output_size_list_.push_back(atom_numbers * sizeof(T)); in InitSizeLists()
88 output_size_list_.push_back(atom_numbers * sizeof(T)); in InitSizeLists()
89 output_size_list_.push_back(3 * atom_numbers * sizeof(T)); in InitSizeLists()
99 std::vector<size_t> output_size_list_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | argmaxandminwithvalue_gpu_kernel.h | 35 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
103 output_size_list_.clear(); in ResetResource()
110 output_size_list_.push_back(output_size_); in InitSizeLists()
111 output_size_list_.push_back(output_size_ / sizeof(S) * sizeof(T)); in InitSizeLists()
119 std::vector<size_t> output_size_list_; variable
|
| D | zeroslike_gpu_kernel.h | 33 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
76 output_size_list_.clear(); in ResetResource()
84 output_size_list_.push_back(input_size_ * sizeof(T)); in InitSizeLists()
91 std::vector<size_t> output_size_list_; variable
|
| D | unique_gpu_kernel.h | 33 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
98 output_size_list_.clear(); in ResetResource()
105 output_size_list_.push_back(output_size_); in InitSizeLists()
106 output_size_list_.push_back(num_elements_ * sizeof(S)); in InitSizeLists()
120 std::vector<size_t> output_size_list_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | nms_with_mask_gpu_kernel.h | 44 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
107 output_size_list_.push_back(num_input_ * sizeof(T) * box_size_); in InitSizeLists()
108 output_size_list_.push_back(num_input_ * sizeof(int)); in InitSizeLists()
109 output_size_list_.push_back(num_input_ * sizeof(bool)); in InitSizeLists()
128 std::vector<size_t> output_size_list_; variable
|
| D | logical_not_gpu_kernel.h | 37 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
69 output_size_list_.clear(); in ResetResource()
76 output_size_list_.push_back(input_num_ * sizeof(T)); in InitSizeLists()
83 std::vector<size_t> output_size_list_; variable
|
| D | square_sum_all_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
71 output_size_list_.push_back(sizeof(T)); in InitSizeLists()
72 output_size_list_.push_back(sizeof(T)); in InitSizeLists()
79 std::vector<size_t> output_size_list_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
| D | sync_batch_norm_gpu_kernel.h | 39 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList()
177 output_size_list_.clear(); in ResetResource()
188 output_size_list_.push_back(output_size_); // output in InitSizeLists()
189 output_size_list_.push_back(param_size_S_); // save scale in InitSizeLists()
190 output_size_list_.push_back(param_size_S_); // reserve space in InitSizeLists()
191 output_size_list_.push_back(param_size_G_output_); // save mean in InitSizeLists()
192 output_size_list_.push_back(param_size_G_output_); // save variance in InitSizeLists()
236 std::vector<size_t> output_size_list_; variable
|