Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Didentity.cc84 auto workspaces = PyBoostUtils::GetKernelTensorFromAddress(workspace_address); in IdentityCustomizeCallWithoutContigous() local
132 auto workspaces = PyBoostUtils::GetKernelTensorFromAddress(workspace_address); in IdentityCustomizeCall() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dtopk_cpu_kernel.cc33 const std::vector<KernelTensor *> &workspaces, in LaunchKernel()
124 const std::vector<kernel::KernelTensor *> &workspaces, in Launch()
Duniform_candidate_sampler_cpu_kernel.h47 …ool Launch(const std::vector<KernelTensor *> &inputs, const std::vector<KernelTensor *> &workspace… in Launch()
Dbincount_cpu_kernel.cc86 const std::vector<KernelTensor *> &workspaces, in Launch()
Dfused_ada_factor_cpu_kernel.cc205 const std::vector<KernelTensor *> &workspaces, in LaunchKernel()
288 const std::vector<kernel::KernelTensor *> &workspaces, in Launch()
Duniform_candidate_sampler_cpu_kernel.cc228const std::vector<KernelTensor *> &workspaces, in LaunchKernel()
Ddeformable_offsets_cpu_kernel.cc255 const std::vector<KernelTensor *> &workspaces, in LaunchKernel()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/kernel_packet/
Dkernel_packet_kernel_mod.cc151 const std::vector<KernelTensor *> &workspaces, in Launch()
171const std::vector<KernelTensor *> &workspaces, in GetLaunchArgs()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/debug/
Dassert_gpu_kernel.h36 …ool Launch(const std::vector<KernelTensor *> &inputs, const std::vector<KernelTensor *> &workspace… in Launch()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/
Dcontext.cc64 const std::vector<size_t> &CoderContext::workspaces() const { return dynamic_mem_manager_->GetWorkS… in workspaces() function in mindspore::lite::micro::CoderContext
/third_party/rust/rust/src/tools/rust-analyzer/crates/rust-analyzer/src/
Dreload.rs206 let mut workspaces = linked_projects in fetch_workspaces() localVariable
265 let workspaces = Arc::clone(&self.workspaces); in fetch_build_data() localVariable
356 let workspaces = in switch_workspaces() localVariable
370 let workspaces = workspaces in switch_workspaces() localVariable
Dglobal_state.rs114 pub(crate) workspaces: Arc<Vec<ProjectWorkspace>>, field
134 pub(crate) workspaces: Arc<Vec<ProjectWorkspace>>, field
/third_party/rust/rust/src/tools/rustfmt/src/cargo-fmt/test/
Dtargets.rs75 mod workspaces { module
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dtopk_gpu_kernel.h39 …ool Launch(const std::vector<KernelTensor *> &inputs, const std::vector<KernelTensor *> &workspace… in Launch()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dlamb_gpu_kernel.h65 …ool Launch(const std::vector<KernelTensor *> &inputs, const std::vector<KernelTensor *> &workspace… in Launch()
264 …void CalcTrustRatio(const std::vector<KernelTensor *> &workspaces, float *var_float, float *grad_f… in CalcTrustRatio()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/
Dbuffer_sample_gpu_kernel.cc85 const std::vector<KernelTensor *> &workspaces, in Launch()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/
Drandom_choice_with_mask_gpu_kernel.cc84 const std::vector<KernelTensor *> &workspaces, in LaunchKernel()
Drandom_categorical_gpu_kernel.h40 …ool Launch(const std::vector<KernelTensor *> &inputs, const std::vector<KernelTensor *> &workspace… in Launch()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/kash/
Dkernel_pack.h60 std::vector<size_t> workspaces; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/
Dbisheng_kernel_mod.cc154 const std::vector<AddressPtr> &workspaces, in GenTask()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_runner.cc381 std::vector<kernel::KernelTensor *> workspaces; in GetWorkspaceKernelTensors() local
436 std::vector<kernel::KernelTensor *> workspaces = in GetWorkspaceKernelTensors() local
463 std::vector<kernel::KernelTensor *> workspaces; in GetWorkspaceKernelTensorsDynamic() local
611 …auto workspaces = GetWorkspaceKernelTensors(runtime_info, device_context, node, is_dynamic_shape, … in LaunchKernels() local
/third_party/parse5/
Dpackage.json5 "workspaces": [ array
/third_party/rust/rust/src/tools/rust-analyzer/editors/code/src/
Dctx.ts212 const workspaces: JsonProject[] = await Promise.all( constant
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dascend_kernel_runtime.cc92 …AnfNodePtr &kernel, const kernel::KernelMod *kernel_mod, std::vector<KernelTensor *> *workspaces) { in TbeLaunchKernelModRegister()
518 std::vector<KernelTensor *> workspaces; in RunDynamicKernelAsync() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_kernel_runtime.cc417 std::vector<kernel::KernelTensor *> *workspaces) { in GetRuntimeAddressFromNode()

12