Searched defs:SetInputSizeList (Results 1 – 4 of 4) sorted by relevance
34 void SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = size_list; } in SetInputSizeList() function
40 void AkgKernelMod::SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = size… in SetInputSizeList() function in mindspore::kernel::AkgKernelMod
90 void GpuKernelMod::SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = size… in SetInputSizeList() function in mindspore::kernel::GpuKernelMod
55 void AicpuOpKernelMod::SetInputSizeList(const std::vector<size_t> &size_list) { input_size_list_ = … in SetInputSizeList() function in mindspore::kernel::AicpuOpKernelMod