Searched defs:k_type (Results 1 – 8 of 8) sorted by relevance
62 auto k_type = input_args[kIndexK]->GetType(); in InferType() local
86 auto k_type = input_args[kInputIndex1]->GetType(); in TopKInferType() local
128 auto k_type = input_args[kInputIndex1]->GetType(); in MatrixDiagPartV3InferType() local
149 auto k_type = k_arg->GetType(); in MatrixSetDiagV3InferType() local
197 auto k_type = input_args[kInputIndex1]->GetType(); in MatrixDiagV3InferType() local
70 auto k_type = input_args[kIndexK]->GetType(); in InferType() local
34 auto k_type = input_args[kInputIndex1]->GetType(); in InferType() local
44 TypePtr k_type; member