Searched refs:SetInputSizeList (Results 1 – 11 of 11) sorted by relevance
41 kernel_mod_ptr->SetInputSizeList(json_generator.input_size_list()); in AkgSetKernelMod()
32 void SetInputSizeList(const std::vector<size_t> &size_list);
40 void AkgKernelMod::SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = size… in SetInputSizeList() function in mindspore::kernel::AkgKernelMod
43 kernel_mod_ptr->SetInputSizeList(json_generator.input_size_list()); in AkgSetKernelMod()
61 void SetInputSizeList(const std::vector<size_t> &size_list);
90 void GpuKernelMod::SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = size… in SetInputSizeList() function in mindspore::kernel::GpuKernelMod
34 void SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = size_list; } in SetInputSizeList() function
293 kernel_mod_ptr->SetInputSizeList(input_size_list); in GenKernelMod()
51 void SetInputSizeList(const std::vector<size_t> &size_list);
55 void AicpuOpKernelMod::SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = … in SetInputSizeList() function in mindspore::kernel::AicpuOpKernelMod
88 kernel_mod_ptr->SetInputSizeList(input_size_list); in SetIOSize()