Searched refs:GetAllInputDeviceTypes (Results 1 – 16 of 16) sorted by relevance
29 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(matmul_node); in UpdateBuildInfo()31 auto cast_types = AnfAlgo::GetAllInputDeviceTypes(cast_node); in UpdateBuildInfo()
173 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(matmul); in InsertPad()225 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(matmul_node); in UpdateMatmulInfo()
176 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(cnode); in ReplaceAssignByInplaceAssignInGraphkernel()
152 auto origin_inputs_type = origin_kernel_build_info->GetAllInputDeviceTypes(); in CorrectKernelBuildInfo()
69 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(cnode); in UpdateOutputInfo()
248 auto ori_input_format = AnfAlgo::GetAllInputDeviceTypes(node); in Process_Son()
244 auto all_inputs_type = AnfAlgo::GetAllInputDeviceTypes(node); in CanExpand()
239 auto origin_inputs_type = origin_kernel_build_info->GetAllInputDeviceTypes(); in CorrectKernelBuildInfo()
50 new_transdata_builder->SetInputsDeviceType(transdata_kernel_build_info->GetAllInputDeviceTypes()); in Process()
67 split_new_builder->SetInputsDeviceType(split_kernel_build_info->GetAllInputDeviceTypes()); in Process()
77 const std::vector<TypeId> &GetAllInputDeviceTypes() const;
60 const std::vector<TypeId> &KernelBuildInfo::GetAllInputDeviceTypes() const { return inputs_device_t… in GetAllInputDeviceTypes() function in mindspore::kernel::KernelBuildInfo
143 static std::vector<TypeId> GetAllInputDeviceTypes(const AnfNodePtr &node);
688 std::vector<TypeId> AnfRuntimeAlgorithm::GetAllInputDeviceTypes(const AnfNodePtr &node) { in GetAllInputDeviceTypes() function in mindspore::session::AnfRuntimeAlgorithm699 auto types = build_info->GetAllInputDeviceTypes(); in GetAllInputDeviceTypes()
87 auto supported_akg_type = kernel_info_list[i]->GetAllInputDeviceTypes(); in SupportedTypeList()
559 builder->SetInputsDeviceType(kernel_build_info->GetAllInputDeviceTypes()); in SetKernelInfo()