Searched defs:input_dtypes (Results 1 – 12 of 12) sorted by relevance
28 std::vector<TypePtr> input_dtypes = {kFloat16, kFloat16}; in GetBinaryOpDefaultCases() local
37 std::vector<TypePtr> input_dtypes = {kFloat16, kFloat16, kFloat16, kFloat16}; in GetCases() local
26 std::vector<TypeId> input_dtypes = {dtype_, dtype_}; in ObtainMulKernelGraph() local
24 …const std::string &op_name, const std::vector<TypeId> &input_dtypes, const std::vector<ShapeVector… in ConstructKernelGraphBasedOnSingleOp()
42 std::vector<TypeId> input_dtypes; member
135 std::string input_dtypes; in InitInputOptions() local
60 std::vector<std::string> input_dtypes; member
83 std::vector<TypeId> input_dtypes = {dtype_}; in ConstructKernelGraph() local
41 std::vector<TypeId> input_dtypes; member
123 std::string input_dtypes; member
74 auto input_dtypes = inputs_ptr[j]->dtypes(); in GetSupportFromOpLib() local
883 const auto &input_dtypes = kernel_build_info->GetAllInputDeviceTypes(); in UnfoldKernelBuildInfo() local