| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_delegate_fp32.h | 96 size_t workspace_size() override { in workspace_size() function
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_delegate_fp16.h | 58 size_t workspace_size() override { in workspace_size() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | cholesky_trsm_solve_gpu_kernel.h | 104 size_t workspace_size = batch_ * sizeof(T *); in InitSizeLists() local
116 size_t workspace_size = batch_ * sizeof(T *); in InitSizeLists() local
|
| D | cholesky_solve_gpu_kernel.h | 239 size_t workspace_size; in InitSizeLists() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | dropout3d_gpu_kernel.h | 140 size_t workspace_size = num_chan_ * sizeof(float); // rand_f for curandGen in InitSizeLists() local
|
| D | bias_add_grad_gpu_kenel.h | 200 size_t indices_size, workspace_size; in InitSizeLists() local
|
| D | lstm_grad_weight_gpu_kernel.h | 202 size_t workspace_size = 0; in InitSizeLists() local
|
| D | lstm_gpu_kernel.h | 254 size_t workspace_size = 0; in InitSizeLists() local
|
| D | lstm_grad_data_gpu_kernel.h | 252 size_t workspace_size = 0; in InitSizeLists() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | hccl_adapter.h | 38 int64_t workspace_size; member
|
| D | converter.cc | 223 int64_t workspace_size = workspace_sizes[0]; in ParseDomiTask() local
|
| D | hccl_adapter.cc | 258 int64_t workspace_size = workspace_sizes[0]; in CalcWorkspaceSize() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | batchnorm_fold2_grad_gpu_kernel.h | 136 size_t workspace_size = batch_size_ * channel_ * sizeof(T); in InitSizeLists() local
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | inner_kernel.h | 161 virtual size_t workspace_size() { return workspace_size_; } in workspace_size() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 232 … void *output_data_addr, void *workspace_addr, int64_t workspace_size, int64_t hccl_stream_num, in HcclTaskInfo()
255 int64_t workspace_size() const { return workspace_size_; } in workspace_size() function
|
| /third_party/mindspore/mindspore/lite/src/c_api/ |
| D | model_c.cc | 308 void OH_AI_ModelSetWorkspace(OH_AI_ModelHandle model, void *workspace, size_t workspace_size) { in OH_AI_ModelSetWorkspace()
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_session.cc | 96 size_t workspace_size = 0; in AllocWorkSpace() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 386 …size_t workspace_size = std::accumulate(workspace_size_list.begin(), workspace_size_list.end(), kZ… in FetchMemUnitSize() local
|
| /third_party/pcre2/pcre2/src/ |
| D | pcre2_intmodedep.h | 726 PCRE2_SIZE workspace_size; /* Size of workspace */ member
|
| /third_party/glib/glib/pcre/ |
| D | pcre_internal.h | 2033 int workspace_size; /* Size of workspace */ member
|