Searched refs:SetOutputSizeList (Results 1 – 11 of 11) sorted by relevance
42 kernel_mod_ptr->SetOutputSizeList(json_generator.output_size_list()); in AkgSetKernelMod()
33 void SetOutputSizeList(const std::vector<size_t> &size_list);
42 void AkgKernelMod::SetOutputSizeList(const std::vector<size_t> &size_list) { output_size_list_ = si… in SetOutputSizeList() function in mindspore::kernel::AkgKernelMod
44 kernel_mod_ptr->SetOutputSizeList(json_generator.output_size_list()); in AkgSetKernelMod()
62 void SetOutputSizeList(const std::vector<size_t> &size_list);
92 void GpuKernelMod::SetOutputSizeList(const std::vector<size_t> &size_list) { output_size_list_ = si… in SetOutputSizeList() function in mindspore::kernel::GpuKernelMod
35 void SetOutputSizeList(const std::vector<size_t> &size_list) { output_size_list_ = size_list; } in SetOutputSizeList() function
294 kernel_mod_ptr->SetOutputSizeList(output_size_list); in GenKernelMod()
52 void SetOutputSizeList(const std::vector<size_t> &size_list);
57 void AicpuOpKernelMod::SetOutputSizeList(const std::vector<size_t> &size_list) { output_size_list_ … in SetOutputSizeList() function in mindspore::kernel::AicpuOpKernelMod
101 kernel_mod_ptr->SetOutputSizeList(output_size_list); in SetIOSize()