| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/ |
| D | dvpp_image_utils.cc | 131 void *workspace_addr = nullptr; in DvppAdjustBrightness() local
221 void *workspace_addr = nullptr; in DvppAdjustContrast() local
311 void *workspace_addr = nullptr; in DvppAdjustHue() local
401 void *workspace_addr = nullptr; in DvppAdjustSaturation() local
492 void *workspace_addr = nullptr; in DvppAdjustSharpness() local
609 void *workspace_addr = nullptr; in DvppAffine() local
729 void *workspace_addr = nullptr; in DvppAutoContrast() local
907 void *workspace_addr = nullptr; in DvppConvertColor() local
1002 void *workspace_addr = nullptr; in DvppCrop() local
1133 void *workspace_addr = nullptr; in DvppErase() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/ |
| D | op_api_exec.h | 293 #define RUN_OP_API_ASYNC(aclnn_api, workspace_addr, workspace_size, executor, acl_stream, release_f… argument
310 #define RUN_OP_API_SYNC(aclnn_api, workspace_addr, workspace_size, executor, acl_stream) … argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | random_truncatednorm_gpu_kernel.cc | 82 void *workspace_addr = GetDeviceAddress<void *>(workspace, 0); in LaunchKernel() local
|
| D | non_deterministic_ints_gpu_kernel.cc | 63 void *workspace_addr = GetDeviceAddress<void *>(workspace, 0); in LaunchKernel() local
|
| D | random_op_gpu_kernel.cc | 86 void *workspace_addr = GetDeviceAddress<void *>(workspace, 0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/ |
| D | random_standard_laplace_gpu_kernel.cc | 91 void *workspace_addr = GetDeviceAddress<void *>(workspace, 0); in LaunchKernel() local
|
| D | random_poisson_gpu_kernel.cc | 79 void *workspace_addr = GetDeviceAddress<void *>(workspace, 0); in LaunchKernel() local
|
| D | random_shuffle_gpu_kernel.cc | 115 auto *workspace_addr = GetDeviceAddress<int>(workspace, 0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | softmax_cross_entropy_with_logits_gpu_kernel.cc | 80 T *workspace_addr = GetDeviceAddress<T>(workspace, 0); in LaunchKernel() local
|
| D | lstm_grad_weight_gpu_kernel.cc | 142 void *workspace_addr = GetDeviceAddress<T>(workspace, 0); in LaunchKernel() local
|
| D | dense_grad_gpu_kernel.cc | 159 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, kIndex1); in LaunchKernel() local
|
| D | lstm_gpu_kernel.cc | 160 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in LaunchKernel() local
|
| D | l2normalize_grad_gpu_kernel.cc | 104 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, kIndex2); in LaunchKernel() local
|
| D | lstm_grad_data_gpu_kernel.cc | 158 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in LaunchKernel() local
|
| D | bias_add_grad_gpu_kenel.cc | 95 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, kIndex1); in LaunchKernel() local
|
| D | instance_norm_gpu_kernel.cc | 130 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, kIndex4); in LaunchKernel() local
|
| D | instance_norm_grad_gpu_kernel.cc | 135 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, kIndex3); in LaunchKernel() local
|
| D | batch_norm_grad_gpu_kernel.cc | 196 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, kIndex0); in LaunchKernel() local
|
| D | batch_norm_gpu_kernel.cc | 53 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, kIndex0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/convolution/ |
| D | conv2d_cudnn_gpu_kernel.h | 652 void *workspace_addr = nullptr; in LaunchForward() local
685 void *workspace_addr = nullptr; in LaunchInputGrad() local
718 void *workspace_addr = nullptr; in LaunchFilterGrad() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | smooth_l1_loss_cpu_kernel.cc | 69 T *workspace_addr = reinterpret_cast<T *>(workspace[0]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/ |
| D | statistic_kernel.cc | 148 …auto workspace_addr = GetWorkSpaceDeviceAddress(stream_id, {input}, {output_addr->kernel_tensor().… in LaunchKernel() local
186 …auto workspace_addr = GetWorkSpaceDeviceAddress(stream_id, inputs, {output_addr->kernel_tensor().g… in Launch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/ |
| D | gru_grad_weight_gpu_kernel.h | 73 void *workspace_addr = GetDeviceAddress<T>(workspace, 0); in Launch() local
|
| D | dynamic_rnn_op_gpu_kernel.cc | 165 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/ |
| D | send_actor.cc | 145 auto workspace_addr = send_workspace[kIndex0]; in BuildRpcMessage() local
|