Home
last modified time | relevance | path

Searched defs:workspaces (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dtopk_cpu_kernel.cc30 …nel::LaunchKernel(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in LaunchKernel()
113 const std::vector<kernel::AddressPtr> &workspaces, in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dl2_loss_gpu_kernel.h36 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/rl/
Dbuffer_sample_gpu_kernel.cc91 …pleKernel::Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dtopk_gpu_kernel.h40 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/random/
Drandom_categorical_gpu_kernel.h39 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
Drandom_choice_with_mask_gpu_kernel.h40 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
Duniform_candidate_sampler_gpu_kernel.h48 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/neighbor_list/
Dneighbor_list_update_new_kernel.h64 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
Dneighbor_list_update_kernel.h63 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspaces, in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel.cc277 AddressPtrList workspaces; in GenDynamicKernel() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.cc74 const std::vector<AddressPtr> &workspaces, in GenTask()
/third_party/typescript_eslint/
Dpackage.json4 "workspaces": [ array
/third_party/mindspore/mindspore/ccsrc/ps/
Dparameter_server.cc294 const std::vector<kernel::AddressPtr> &workspaces = optim_info->workspaces(); in UpdateWeights() local
412 std::vector<kernel::AddressPtr> workspaces; in DoEmbeddingLookup() local
Doptimizer_info.cc33 const std::vector<AddressPtr> &OptimizerInfo::workspaces() const { return workspaces_; } in workspaces() function in mindspore::ps::OptimizerInfo
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel.h125 std::vector<size_t> workspaces; member
/third_party/weston/desktop-shell/
Dshell.h192 } workspaces; member
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc830 … const AddressPtrList &inputs, const AddressPtrList &workspaces, in LaunchKernelWithoutMock()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc904 auto workspaces = somas_node->workspace_tensors_; in UnReuseNodeProcess() local