Home
last modified time | relevance | path

Searched defs:input_size_list_ (Results 1 – 25 of 258) sorted by relevance

1234567891011

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_mod.h45 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/rl/
Dbuffer_get_gpu_kernel.h48 std::vector<size_t> input_size_list_; variable
Dbuffer_append_gpu_kernel.h49 std::vector<size_t> input_size_list_; variable
Dbuffer_sample_gpu_kernel.h56 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/
Ddataset_init_kernel.h47 std::vector<size_t> input_size_list_; variable
Ddataset_iterator_kernel.h53 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/
Dminmax_update_perlayer_gpu_kernel.h43 std::vector<size_t> input_size_list_; variable
Dfake_learned_scale_quant_perlayer_gpu_kernel.h43 std::vector<size_t> input_size_list_; variable
Dfake_learned_scale_quant_perlayer_grad_gpu_kernel.h44 std::vector<size_t> input_size_list_; variable
Dminmax_update_perchannel_gpu_kernel.h43 std::vector<size_t> input_size_list_; variable
Dfake_learned_scale_quant_perchannel_gpu_kernel.h43 std::vector<size_t> input_size_list_; variable
Dfake_learned_scale_quant_perchannel_grad_gpu_kernel.h44 std::vector<size_t> input_size_list_; variable
Dfake_quant_perchannel_grad_gpu_kernel.h43 std::vector<size_t> input_size_list_; variable
Dfake_quant_perlayer_grad_gpu_kernel.h44 std::vector<size_t> input_size_list_; variable
Dfake_quant_perlayer_gpu_kernel.h43 std::vector<size_t> input_size_list_; variable
Dfake_quant_perchannel_gpu_kernel.h46 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/
Dtrt_kernel.h49 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.h68 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.h50 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/
Drecv_gpu_kernel.h62 std::vector<size_t> input_size_list_; variable
Dsend_gpu_kernel.h62 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_mod.h43 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/
Dakg_gpu_kernel_mod.h74 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dlogical_not_gpu_kernel.h82 std::vector<size_t> input_size_list_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dzeroslike_gpu_kernel.h90 std::vector<size_t> input_size_list_; variable

1234567891011