Searched refs:SetInputsReshapeType (Results 1 – 9 of 9) sorted by relevance
54 builder.SetInputsReshapeType({""}); in SelectKernel()63 builder.SetInputsReshapeType({""}); in SelectKernel()82 builder.SetInputsReshapeType({""}); in SelectKernel()91 builder.SetInputsReshapeType({""}); in SelectKernel()128 builder.SetInputsReshapeType({""}); in TEST_F()176 builder.SetInputsReshapeType({""}); in TEST_F()
61 builder.SetInputsReshapeType({""}); in SelectKernel()70 builder.SetInputsReshapeType({""}); in SelectKernel()100 builder.SetInputsReshapeType({""}); in TEST_F()
53 builder.SetInputsReshapeType({"", ""}); in GetSingleOutputGraph()75 builder.SetInputsReshapeType({""}); in GetMutilpleOutputGraph()95 builder.SetInputsReshapeType({""}); in SelectKernel()
56 builder.SetInputsReshapeType({"", ""}); in GetSingleOutputGraph()83 builder.SetInputsReshapeType({""}); in GetMutilpleOutputGraph()106 builder.SetInputsReshapeType({""}); in SelectKernel()
59 builder.SetInputsReshapeType({""}); in SelectKernel()107 builder.SetInputsReshapeType({"", ""}); in TEST_F()174 builder.SetInputsReshapeType({"", ""}); in TEST_F()251 builder.SetInputsReshapeType({"", ""}); in TEST_F()
161 void SetInputsReshapeType(const std::vector<std::string> &input_reshape_type);
193 void KernelBuildInfo::KernelBuildInfoBuilder::SetInputsReshapeType(const std::vector<std::string> &… in SetInputsReshapeType() function in mindspore::kernel::KernelBuildInfo::KernelBuildInfoBuilder
251 builder->SetInputsReshapeType({reshape_type}); in RefreshKernelBuildInfo()339 builder.SetInputsReshapeType({reshape_type}); in AddCastOpNodeToGraph()
119 builder.SetInputsReshapeType(inputs_reshape_type); in GetCommonPatternKernelInfo()