Searched refs:InferExpandDimsType (Results 1 – 1 of 1) sorted by relevance
116 Type InferExpandDimsType(Type ty, int64_t axis, Builder *builder) { in InferExpandDimsType() function762 inferred_ty = InferExpandDimsType(input_ty, axis, &rewriter); in matchAndRewrite()