Home
last modified time | relevance | path

Searched defs:workspace_size_list_ (Results 1 – 25 of 287) sorted by relevance

12345678910>>...12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_mod.h47 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/rl/
Dbuffer_get_gpu_kernel.h50 std::vector<size_t> workspace_size_list_; variable
Dbuffer_append_gpu_kernel.h51 std::vector<size_t> workspace_size_list_; variable
Dbuffer_sample_gpu_kernel.h58 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/
Ddataset_init_kernel.h49 std::vector<size_t> workspace_size_list_; variable
Ddataset_iterator_kernel.h55 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/
Dminmax_update_perlayer_gpu_kernel.h45 std::vector<size_t> workspace_size_list_; variable
Dfake_learned_scale_quant_perlayer_gpu_kernel.h45 std::vector<size_t> workspace_size_list_; variable
Dfake_learned_scale_quant_perlayer_grad_gpu_kernel.h46 std::vector<size_t> workspace_size_list_; variable
Dminmax_update_perchannel_gpu_kernel.h45 std::vector<size_t> workspace_size_list_; variable
Dfake_learned_scale_quant_perchannel_grad_gpu_kernel.h46 std::vector<size_t> workspace_size_list_; variable
Dfake_learned_scale_quant_perchannel_gpu_kernel.h45 std::vector<size_t> workspace_size_list_; variable
Dfake_quant_perchannel_grad_gpu_kernel.h45 std::vector<size_t> workspace_size_list_; variable
Dfake_quant_perlayer_grad_gpu_kernel.h46 std::vector<size_t> workspace_size_list_; variable
Dfake_quant_perlayer_gpu_kernel.h45 std::vector<size_t> workspace_size_list_; variable
Dfake_quant_perchannel_gpu_kernel.h48 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/
Dtrt_kernel.h51 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.h70 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Drt_kernel.h42 mutable std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.h52 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/
Drecv_gpu_kernel.h64 std::vector<size_t> workspace_size_list_; variable
Dsend_gpu_kernel.h64 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_mod.h45 std::vector<size_t> workspace_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/
Dmd_iteration_gradient_descent_kernel.h71 std::vector<size_t> workspace_size_list_; variable
Dmd_iteration_setup_random_state.h67 std::vector<size_t> workspace_size_list_; variable

12345678910>>...12