Searched refs:expand_shape (Results 1 – 2 of 2) sorted by relevance
268 SmallVector<int64_t, 4> expand_shape(input_shape.begin(), in matchAndRewrite()270 expand_shape.insert(expand_shape.begin() + expand_axis, 1); in matchAndRewrite()273 expand_shape, getElementTypeOrSelf(stb_op.input())); in matchAndRewrite()
230 SmallVector<int64_t, 4> expand_shape(pad_shape.begin(), pad_shape.end()); in GetPadOpForConv2DFilter() local233 RankedTensorType::get(expand_shape, getElementTypeOrSelf(filter)); in GetPadOpForConv2DFilter()