| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_all_to_all.cc | 95 void *workspace_addr = nullptr; in GenTask() local
|
| D | hccl_kernel.cc | 256 void *workspace_addr = nullptr; in GenTask() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | multinomial_gpu_kernel.h | 54 void *workspace_addr = GetDeviceAddress<void *>(workspace, 1); in Launch() local
|
| D | random_op_gpu_kernel.h | 71 void *workspace_addr = GetDeviceAddress<void *>(workspace, 0); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | bias_add_grad_gpu_kenel.h | 66 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 1); in Launch() local
|
| D | instance_norm_gpu_kernel.h | 78 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 4); in Launch() local
|
| D | l2normalize_gpu_kernel.h | 63 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 1); in Launch() local
|
| D | lstm_grad_weight_gpu_kernel.h | 68 void *workspace_addr = GetDeviceAddress<T>(workspace, 0); in Launch() local
|
| D | instance_norm_grad_gpu_kernel.h | 79 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 3); in Launch() local
|
| D | batch_norm_gpu_kernel.h | 59 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
|
| D | lstm_gpu_kernel.h | 77 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
|
| D | batch_norm_grad_gpu_kernel.h | 96 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
|
| D | l2normalize_grad_gpu_kernel.h | 67 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 2); in Launch() local
|
| D | lstm_grad_data_gpu_kernel.h | 81 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
|
| D | conv3d_gpu_kernel.h | 48 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
|
| D | conv2d_gpu_kernel.h | 49 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | array_reduce_gpu_kernel.h | 52 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | rolling_cpu_kernel.cc | 161 auto workspace_addr = reinterpret_cast<size_t *>(workspace[0]->addr); in Launch() local
|
| /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()
254 void *workspace_addr() const { return workspace_addr_; } in workspace_addr() function
|