Home
last modified time | relevance | path

Searched defs:block_dim (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dai_core_dynamic_kernel.h35 …AiCoreDynamicKernel(const void *stub_fubc, uint32_t block_dim, void *tiling_data_ptr, uint32_t op_… in AiCoreDynamicKernel()
43 …AiCoreDynamicKernel(void *handle, uint32_t block_dim, void *tiling_data_ptr, uint32_t op_para_size… in AiCoreDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dascend_kernel_mod.h35 uint32_t block_dim() { return block_dim_; } in block_dim() function
Dkernel.h120 uint32_t block_dim; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_mod.cc66 uint32_t block_dim = DEFAULT_BLOCK_DIM; // default blockdim equal to 1. in Launch() local
122 uint32_t block_dim = DEFAULT_BLOCK_DIM; // default blockdim equal to 1. in GenTask() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dprofiling_desc.h39 TaskDesc(std::string op_name, uint32_t task_id, uint32_t block_dim, uint32_t stream_id) in TaskDesc()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.cc145 uint32_t block_dim = 1; // default blockdim equal to 1. in GenDynamicKernel() local
Dtbe_utils.cc278 uint32_t *block_dim, const bool dynamic_flag, void **handle, in GenFuncStub()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/
Dtask_info.h68 …o(const std::string &op_name, uint32_t stream_id, const std::string &stub_func, uint32_t block_dim, in TbeTaskInfo()
87 uint32_t block_dim() const { return block_dim_; } in block_dim() function