Searched refs:op_para_size (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_core_dynamic_kernel.h | 35 …amicKernel(const void *stub_fubc, uint32_t block_dim, void *tiling_data_ptr, uint32_t op_para_size, in AiCoreDynamicKernel() argument
41 op_para_size_(op_para_size), in AiCoreDynamicKernel()
43 …ernel(void *handle, uint32_t block_dim, void *tiling_data_ptr, uint32_t op_para_size, void *stream, in AiCoreDynamicKernel() argument
49 op_para_size_(op_para_size), in AiCoreDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 122 auto op_para_size = kernel_json_info.op_para_size; in GenDynamicKernel() local
136 if (op_para_size > 0) { in GenDynamicKernel()
137 auto ret = rtMalloc(&tiling_data_ptr, op_para_size, RT_MEMORY_HBM); in GenDynamicKernel()
154 …:make_shared<device::ascend::AiCoreDynamicKernel>(handle, block_dim, tiling_data_ptr, op_para_size, in GenDynamicKernel()
162 … op_para_size, stream_ptr, cnode_ptr, runtime_args); in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel.h | 126 uint32_t op_para_size; member
127 KernelJsonInfo() : block_dim(0), op_para_size(0) {} in KernelJsonInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/kash/ |
| D | kernel_pack.cc | 179 kernel_json_info_.op_para_size = js["opParaSize"]; in ParseKernelJson()
|