Searched refs:matmul_shape (Results 1 – 2 of 2) sorted by relevance
325 std::vector<int64_t> matmul_shape; in rewriteToBatchMatmul() local327 &matmul_shape, &rewriter))) in rewriteToBatchMatmul()335 RankedTensorType::get(matmul_shape, original_type.getElementType()); in rewriteToBatchMatmul()
526 llvm::SmallVector<int64_t, 4> matmul_shape = in ConvertDot() local533 loc, RankedTensorType::get(matmul_shape, result_type.getElementType()), in ConvertDot()