Home
last modified time | relevance | path

Searched defs:InitResource (Results 1 – 25 of 45) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Dmulti_node_split.cc24 int MultiNodeSplitProxy::InitResource() { in InitResource() function in mindspore::opt::MultiNodeSplitProxy
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Ddevice_resource.cc22 Status DeviceResource::InitResource(uint32_t) { in InitResource() function in mindspore::dataset::DeviceResource
Dascend_resource.cc27 Status AscendResource::InitResource(uint32_t device_id) { in InitResource() function in mindspore::dataset::AscendResource
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/
DResourceManager.cc98 APP_ERROR ResourceManager::InitResource(ResourceInfo &resourceInfo) { in InitResource() function in ResourceManager
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dbias_add_gpu_kernel.h48 void InitResource() override { in InitResource() function
Dl2normalize_grad_gpu_kernel.h70 void InitResource() override { in InitResource() function
Dlstm_grad_weight_gpu_kernel.h68 void InitResource() override { in InitResource() function
Ddense_grad_gpu_kernel.cc182 void DenseGradGpuKernelMod::InitResource() { in InitResource() function in mindspore::kernel::DenseGradGpuKernelMod
Dlstm_gpu_kernel.h115 void InitResource() override { in InitResource() function
Dbias_add_grad_gpu_kenel.cc160 void BiasAddGradGpuKernelMod::InitResource() { in InitResource() function in mindspore::kernel::BiasAddGradGpuKernelMod
Dlstm_grad_data_gpu_kernel.h89 void InitResource() override { in InitResource() function
Dsparse_softmax_cross_entropy_with_logits_gpu_kernel.h120 void InitResource() override { in InitResource() function
Dlocal_response_norm_gpu_kernel.h170 void InitResource() override { in InitResource() function
Dbatch_norm_grad_gpu_kernel.cc114 void BatchNormGradGpuKernelMod::InitResource() { in InitResource() function in mindspore::kernel::BatchNormGradGpuKernelMod
Dbatch_norm_gpu_kernel.cc204 void BatchNormGpuKernelMod::InitResource() { in InitResource() function in mindspore::kernel::BatchNormGpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/convolution/
Dconv2d_filter_grad_gpu_kernel.cc131 void Conv2dFilterGradGpuKernelMod::InitResource() { in InitResource() function in mindspore::kernel::Conv2dFilterGradGpuKernelMod
Dconv2d_input_grad_gpu_kernel.cc115 void Conv2dInputGradGpuKernelMod::InitResource() { in InitResource() function in mindspore::kernel::Conv2dInputGradGpuKernelMod
Dconv2d_forward_gpu_kernel.cc92 void Conv2dFwdGpuKernelMod::InitResource() { in InitResource() function in mindspore::kernel::Conv2dFwdGpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/quant/
Dbatchnorm_fold2_gpu_kernel.h99 …void InitResource() override { cudnn_handle_ = device::gpu::GPUDeviceManager::GetInstance().GetCud… in InitResource() function
Dbatchnorm_fold_gpu_kernel.h163 void InitResource() override { in InitResource() function
Dbatchnorm_fold2_grad_gpu_kernel.h142 …void InitResource() override { cudnn_handle_ = device::gpu::GPUDeviceManager::GetInstance().GetCud… in InitResource() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/
Drandom_categorical_gpu_kernel.h146 void InitResource() override {} in InitResource() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/
Ddynamic_rnn_op_gpu_kernel.h116 void InitResource() override { in InitResource() function
Dgru_grad_weight_gpu_kernel.h174 void InitResource() override { in InitResource() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/
Dgpu_kernel.h131 virtual void InitResource() {} in InitResource() function

12